diff --git a/tf_adapter/python/npu_bridge/hccl/hccl_ops.py b/tf_adapter/python/npu_bridge/hccl/hccl_ops.py index d226f58aa295c9a5f599ba9a0dda19852830a7ae..974a51274c2bd9931223462fc4b22a7a2548905e 100644 --- a/tf_adapter/python/npu_bridge/hccl/hccl_ops.py +++ b/tf_adapter/python/npu_bridge/hccl/hccl_ops.py @@ -141,7 +141,7 @@ def remote_write(tensorRemote, tensorLocal, data_type): # @param tensorLocal 本端发送内存基地址 # @param offset 进行跳写的步长 def remote_scatter_write(tensorRemote, tensorLocal, offset): - result = gen_hccl_ops.hcom_remote_write( + result = gen_hccl_ops.hcom_remote_scatter_write( remote=tensorRemote, local=tensorLocal, local_offset=offset)