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 21e87064d99172a58ee0a8d3b8107604832b6798..c245d2cb11859808e1f6d66977c551af52b7f794 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()