Skip to content

Commit e8c51e4

Browse files
style: format comments
1 parent 709d447 commit e8c51e4

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

  • infini_train/src/kernels/cuda

infini_train/src/kernels/cuda/comm.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ std::vector<std::shared_ptr<Tensor>> ReduceAddCoalesced(const std::vector<std::v
3636
to_destination_grads[i].push_back(std::make_shared<Tensor>(grads[i][j]->To(destination)));
3737
}
3838
}
39-
// NOTE(zbl): To ensure Profiler works correctly, there should not be any other kernel calls
39+
// NOTE(zbl): To ensure Profiler works correctly, there should not be any other kernel calls
4040
// between GetKernel and kernel.Call, otherwise ProfileContext would be tainted
4141
auto kernel = Dispatcher::Instance().GetKernel({destination.type(), "AccumulateGrad"});
4242
for (int i = 0; i < grads.size(); ++i) {

0 commit comments

Comments
 (0)