From 48a8f748c679514f433119483cdcafcc0adbaa7b Mon Sep 17 00:00:00 2001 From: l30036321 Date: Tue, 5 Dec 2023 20:23:32 +0800 Subject: [PATCH] add api to inplace list --- .../ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py index a98f56fab..6946be6c2 100644 --- a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py +++ b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/common/utils.py @@ -105,7 +105,8 @@ class Const: MAX_SEED_VALUE = 2**32 - 1 - INPLACE_LIST = ["broadcast", "all_reduce", "reduce", "all_gather", "gather", "scatter", "reduce_scatter"] + INPLACE_LIST = ["broadcast", "all_reduce", "reduce", "all_gather", "gather", "scatter", "reduce_scatter", + "_reduce_scatter_base", "_all_gather_base"] class CompareConst: -- Gitee