From c85a191eb88c84ac2eec2000009baebd0dc8dd77 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9C=B1=E4=BF=8A=E8=BE=89?= <11181965+zhujunhui1@user.noreply.gitee.com> Date: Wed, 12 Apr 2023 09:44:34 +0000 Subject: [PATCH] =?UTF-8?q?=E6=B3=A8=E9=87=8A=E7=AE=97=E5=AD=90=E9=BB=91?= =?UTF-8?q?=E5=90=8D=E5=8D=95update=20PyTorch/built-in/nlp/mBART=5FID2372?= =?UTF-8?q?=5Ffor=5FPyTorch/fairseq=5Fcli/generate.py.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: 朱俊辉 <11181965+zhujunhui1@user.noreply.gitee.com> --- .../nlp/mBART_ID2372_for_PyTorch/fairseq_cli/generate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PyTorch/built-in/nlp/mBART_ID2372_for_PyTorch/fairseq_cli/generate.py b/PyTorch/built-in/nlp/mBART_ID2372_for_PyTorch/fairseq_cli/generate.py index 3a8e157965..e9e3f0aed2 100644 --- a/PyTorch/built-in/nlp/mBART_ID2372_for_PyTorch/fairseq_cli/generate.py +++ b/PyTorch/built-in/nlp/mBART_ID2372_for_PyTorch/fairseq_cli/generate.py @@ -60,7 +60,7 @@ def _main(args, output_file): option={} option['ACL_OP_SELECT_IMPL_MODE'] = 'high_performance' option['ACL_OPTYPELIST_FOR_IMPLMODE'] = 'LayerNorm' - option['NPU_FUZZY_COMPILE_BLACKLIST'] = 'MultiHeadAttention,MultiHeadAttentionGrad' + #option['NPU_FUZZY_COMPILE_BLACKLIST'] = 'MultiHeadAttention,MultiHeadAttentionGrad' torch.npu.set_option(option) utils.import_user_module(args) -- Gitee