From f63783e67d43ab412cea668bc11ca38afb5a034d Mon Sep 17 00:00:00 2001 From: Kaiyu Wang Date: Mon, 1 Sep 2025 22:22:11 +0800 Subject: [PATCH] [built-in][nlp][Bert-Squad-ID0470_for_PyTorch]open internal_format to improve performance when in special device --- .../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