From e248f2cfbbfbc02645fb17af97681d7444993ef8 Mon Sep 17 00:00:00 2001 From: xie <1272599165@qq.com> Date: Mon, 15 Jul 2024 09:53:07 +0800 Subject: [PATCH] add INPLACE_LIST function --- debug/accuracy_tools/atat/pytorch/common/utils.py | 3 ++- .../atat/pytorch/hook_module/support_wrap_ops.yaml | 4 +++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/debug/accuracy_tools/atat/pytorch/common/utils.py b/debug/accuracy_tools/atat/pytorch/common/utils.py index aa4a321407..6ec5d6c9e5 100644 --- a/debug/accuracy_tools/atat/pytorch/common/utils.py +++ b/debug/accuracy_tools/atat/pytorch/common/utils.py @@ -195,7 +195,8 @@ class Const: MAX_SEED_VALUE = 2**32 - 1 INPLACE_LIST = ["broadcast", "all_reduce", "reduce", "all_gather", "gather", "scatter", "reduce_scatter", - "_reduce_scatter_base", "_all_gather_base", "all_to_all_single"] + "_reduce_scatter_base", "_all_gather_base", "all_to_all_single", "all_gather_into_tensor", + "reduce_scatter_tensor"] TASK_LIST = ["tensor", "statistics", "overflow_check", "free_benchmark"] LEVEL_LIST = ["L0", "L1", "L2", "mix"] diff --git a/debug/accuracy_tools/atat/pytorch/hook_module/support_wrap_ops.yaml b/debug/accuracy_tools/atat/pytorch/hook_module/support_wrap_ops.yaml index d64c577ff3..d6b4c6896f 100644 --- a/debug/accuracy_tools/atat/pytorch/hook_module/support_wrap_ops.yaml +++ b/debug/accuracy_tools/atat/pytorch/hook_module/support_wrap_ops.yaml @@ -1873,4 +1873,6 @@ distributed: - reduce_scatter - _reduce_scatter_base - _all_gather_base - - all_to_all_single \ No newline at end of file + - all_to_all_single + - all_gather_into_tensor + - reduce_scatter_tensor \ No newline at end of file -- Gitee