From 7c785562e8b273a2218b71a7c4c47631e681fdda Mon Sep 17 00:00:00 2001 From: Pavan Balaji Date: Fri, 15 Dec 2023 00:51:57 -0800 Subject: [PATCH] Fix breakage in Torch-UCC Summary: The OSS PR https://github.com/pytorch/pytorch/pull/115730 broke our internal build for Torch-UCC. This patch fixes it. Reviewed By: kingchc Differential Revision: D52174522 fbshipit-source-id: 6ed49027f5d560c209ba7a96b5d763a9194cd4d0 --- src/torch_ucc_tracing.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/torch_ucc_tracing.cpp b/src/torch_ucc_tracing.cpp index 994592e..1106b42 100644 --- 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;