Fix breakage in Torch-UCC

Summary: The OSS PR pytorch/pytorch#115730 broke our internal build for Torch-UCC. This patch fixes it. Reviewed By: kingchc Differential Revision: D52174522 fbshipit-source-id: 6ed49027f5d560c209ba7a96b5d763a9194cd4d0
facebookresearch · Dec 15, 2023 · 7c78556 · 7c78556
1 parent 13ccc87
commit 7c78556
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/torch_ucc_tracing.cpp b/src/torch_ucc_tracing.cpp
@@ -177,6 +177,8 @@ void CommTraceLogger::recordComms(
       dtype,
       curInSplitSizes_,
       curOutSplitSizes_,
+      0,
+      0,
       world_size);
 
   ++seqnum;