From ba00ce08e45993baff1c3a25c3a663a84b52bda9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=8E=8B=E5=A7=9C=E5=A5=94?= Date: Tue, 12 Apr 2022 11:25:26 +0000 Subject: [PATCH] del option["MM_BMM_ND_ENABLE"] = "enable" --- PyTorch/built-in/nlp/Bert-Squad_ID0470_for_PyTorch/run_squad.py | 1 - 1 file changed, 1 deletion(-) diff --git a/PyTorch/built-in/nlp/Bert-Squad_ID0470_for_PyTorch/run_squad.py b/PyTorch/built-in/nlp/Bert-Squad_ID0470_for_PyTorch/run_squad.py index 21e87064d9..c245d2cb11 100644 --- a/PyTorch/built-in/nlp/Bert-Squad_ID0470_for_PyTorch/run_squad.py +++ b/PyTorch/built-in/nlp/Bert-Squad_ID0470_for_PyTorch/run_squad.py @@ -1299,7 +1299,6 @@ if __name__ == "__main__": option = {} option["ACL_OP_SELECT_IMPL_MODE"] = "high_performance" option["ACL_OPTYPELIST_FOR_IMPLMODE"] = "LayerNorm" - option["MM_BMM_ND_ENABLE"] = "enable" torch.npu.set_option(option) main() dllogger.flush() -- Gitee