From 30f3369da083f70207dbd2bb0911f484edb8c9c2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=8E=8B=E5=87=AF=E5=AE=87?= Date: Mon, 1 Sep 2025 15:00:59 +0000 Subject: [PATCH] open internal_format to improve performance when in special device MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: 王凯宇 --- .../built-in/nlp/Bert-Squad_ID0470_for_PyTorch/run_squad.py | 5 ++++- 1 file changed, 4 insertions(+), 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 c46721df1c..6ca21e29d2 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 @@ -31,7 +31,10 @@ import numpy as np import torch if torch.__version__ >= "1.8": import torch_npu - torch.npu.config.allow_internal_format = False + if torch_npu.npu.utils.get_soc_version() == 103: + torch.npu.config.allow_internal_format = True + else: + torch.npu.config.allow_internal_format = False from torch.utils.data import (DataLoader, RandomSampler, SequentialSampler, TensorDataset) from torch.utils.data.distributed import DistributedSampler -- Gitee