Skip to content

Commit

Permalink
WIP
Browse files Browse the repository at this point in the history
  • Loading branch information
Binyang2014 committed Dec 2, 2024
1 parent 4f8f9fd commit fac8043
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions apps/nccl/src/nccl.cu
Original file line number Diff line number Diff line change
Expand Up @@ -192,7 +192,7 @@ static std::vector<mscclpp::SmChannel> setupSmChannels(ncclComm_t comm,
return channels;
}

static std::pair<std::string, executionPlanInstance> load_execution_plan(const char* filename) {
static std::pair<std::string, executionPlanInstance> loadExecutionPlan(const char* filename) {
std::shared_ptr<mscclpp::ExecutionPlan> plan = std::make_shared<mscclpp::ExecutionPlan>(filename);
std::string collective = plan->collective();
planKey key{plan->minMessageSize(), plan->maxMessageSize(), plan->isInPlace()};
Expand Down Expand Up @@ -404,7 +404,7 @@ NCCL_API ncclResult_t ncclCommInitRank(ncclComm_t* comm, int nranks, ncclUniqueI
for (const auto& entry : std::filesystem::directory_iterator(collectiveDir)) {
if (entry.is_regular_file()) {
std::string filename = entry.path().filename().string();
auto plan = load_execution_plan(entry.path().c_str());
auto plan = loadExecutionPlan(entry.path().c_str());
commPtr->executionPlans[plan.first].push_back(plan.second);
}
}
Expand Down

0 comments on commit fac8043

Please sign in to comment.