Skip to content

Commit b57ce84

Browse files
committed
Lint
1 parent 0e9a309 commit b57ce84

File tree

3 files changed

+6
-3
lines changed

3 files changed

+6
-3
lines changed

csrc/host_ir/evaluator.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -376,7 +376,8 @@ void HostIrEvaluator::handle(Communication* communication) {
376376
communication->type() == CommunicationType::Reduce)
377377
? root_val
378378
: -1;
379-
// For Broadcast and Reduce, non-roots may have no output; use input for cache key
379+
// For Broadcast and Reduce, non-roots may have no output; use input for
380+
// cache key
380381
at::Tensor cache_buffer =
381382
output_tensor.defined() ? output_tensor : input_tensor;
382383
// For Reduce specifically, use the input tensor for the cache key

csrc/multidevice/cuda_p2p.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -914,7 +914,8 @@ void postReduceWithCudaBackend(
914914
cuStreamBatchMemOp(stream, world_size - 1, ops.data(), 0));
915915

916916
// Root launches the ld_reduce kernel
917-
NVF_ERROR(output.defined(), "Output must be defined for reduce on the root");
917+
NVF_ERROR(
918+
output.defined(), "Output must be defined for reduce on the root");
918919
launchMulticastReduceKernel(
919920
reduce_handle->multicastPtr(), output.data_ptr(), size, stream);
920921

tests/cpp/test_multidevice_stream_parallel_type.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -764,7 +764,8 @@ TEST_P(RSMatmulTest, ReduceScatterReduceBased) {
764764
C->axis(1)->parallelize(ParallelType::DIDx);
765765
C->axis(0)->parallelize(ParallelType::Stream);
766766

767-
// Set the matmul's output to be symmetric in order to use NVLS multimem reduce.
767+
// Set the matmul's output to be symmetric in order to use NVLS multimem
768+
// reduce.
768769
if (communicator_backend == CommunicatorBackend::kCuda) {
769770
C_unreduced->setMemoryType(MemoryType::Symmetric);
770771
}

0 commit comments

Comments
 (0)