This commit is contained in:
Binyang Li
2026-06-16 15:58:10 +00:00
parent de91542355
commit f4fbd093db
5 changed files with 86 additions and 82 deletions

View File

@@ -226,7 +226,7 @@ def main():
)
dist.barrier(group=group)
_skip_verify = os.environ.get("MSCCLPP_EP_SKIP_VERIFY","0") in ("1","true","True")
_skip_verify = os.environ.get("MSCCLPP_EP_SKIP_VERIFY", "0") in ("1", "true", "True")
# Validate recv buffer: for each source rank i, the block carries value i.
assert recv_x.dim() == 2 and recv_x.size(1) == hidden
start = 0