From d22e48df7e28a0da648a92c2d2bb5f2d7f7394bb Mon Sep 17 00:00:00 2001 From: zhuobohang Date: Tue, 16 Jul 2024 09:50:12 +0800 Subject: [PATCH] =?UTF-8?q?[built-in][PyTorch][Bert=5FChinese]=20=E5=88=A0?= =?UTF-8?q?=E9=99=A4BMM=E7=8E=AF=E5=A2=83=E5=8F=98=E9=87=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- PyTorch/built-in/nlp/Bert_Chinese_ID3433_for_PyTorch/run_mlm.py | 1 - .../built-in/nlp/Bert_Chinese_ID3433_for_PyTorch/test/env_npu.sh | 1 - 2 files changed, 2 deletions(-) diff --git a/PyTorch/built-in/nlp/Bert_Chinese_ID3433_for_PyTorch/run_mlm.py b/PyTorch/built-in/nlp/Bert_Chinese_ID3433_for_PyTorch/run_mlm.py index 35ce92029a..cd35fda64d 100644 --- a/PyTorch/built-in/nlp/Bert_Chinese_ID3433_for_PyTorch/run_mlm.py +++ b/PyTorch/built-in/nlp/Bert_Chinese_ID3433_for_PyTorch/run_mlm.py @@ -35,7 +35,6 @@ if torch.__version__ >= "1.8": option['ACL_OP_COMPILER_CACHE_MODE'] = "enable" # cache功能启用 option['ACL_OP_COMPILER_CACHE_DIR'] = "./cache" # cache所在的文件夹 -option['MM_BMM_ND_ENABLE'] = 'disable' option['ACL_OP_SELECT_IMPL_MODE'] = "high_performance" option['ACL_OPTYPELIST_FOR_IMPLMODE'] = "LayerNorm" torch.npu.set_option(option) diff --git a/PyTorch/built-in/nlp/Bert_Chinese_ID3433_for_PyTorch/test/env_npu.sh b/PyTorch/built-in/nlp/Bert_Chinese_ID3433_for_PyTorch/test/env_npu.sh index 7f86fa15ed..6e31cbb0f0 100644 --- a/PyTorch/built-in/nlp/Bert_Chinese_ID3433_for_PyTorch/test/env_npu.sh +++ b/PyTorch/built-in/nlp/Bert_Chinese_ID3433_for_PyTorch/test/env_npu.sh @@ -22,7 +22,6 @@ msnpureport -g error -d 6 msnpureport -g error -d 7 -export BMMV2_ENABLE=1 export SCALAR_TO_HOST_MEM=1 #将Host日志输出到串口,0-关闭/1-开启 export ASCEND_SLOG_PRINT_TO_STDOUT=0 -- Gitee