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 3a8e157965d79b0d7f7a1281ca2a1153a0a6f0b5..e9e3f0aed272363775a2d74c1b7cdcc7a386ed41 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)