Unverified Commit 00ce7e31 authored by Ke Bao's avatar Ke Bao Committed by GitHub
Browse files

Fix all gather torch compile (#3992)


Co-authored-by: default avataryizhang2077 <1109276519@qq.com>
parent 50f28f65
......@@ -156,7 +156,7 @@ if supports_custom_op():
direct_register_custom_op(
op_name="reg_all_gather_into_tensor",
op_func=reg_all_gather_into_tensor,
mutates_args=[],
mutates_args=["output"],
fake_impl=reg_all_gather_into_tensor_fake,
)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment