From 27e843a63861ce836cc7ad3c9b1bbf990b23fbf1 Mon Sep 17 00:00:00 2001
From: loomischen <loomischen@foxmail.com>
Date: Fri, 8 Apr 2022 11:14:35 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E7=8E=AF=E5=A2=83=E5=8F=98?=
 =?UTF-8?q?=E9=87=8FSCALAR=5FTO=5FHOST=5FMEM?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../contrib/cv/classification/EfficientNet-B1/test/env_npu.sh   | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/PyTorch/contrib/cv/classification/EfficientNet-B1/test/env_npu.sh b/PyTorch/contrib/cv/classification/EfficientNet-B1/test/env_npu.sh
index 94ab766ff0..2a14bcdf3a 100644
--- a/PyTorch/contrib/cv/classification/EfficientNet-B1/test/env_npu.sh
+++ b/PyTorch/contrib/cv/classification/EfficientNet-B1/test/env_npu.sh
@@ -54,6 +54,8 @@ export DYNAMIC_OP="ADD#MUL"
 export HCCL_WHITELIST_DISABLE=1
 # HCCL默认超时时间120s较少，修改为1800s对齐PyTorch默认设置
 export HCCL_CONNECT_TIMEOUT=1800
+# 对算子add和mul优化，不加可能导致模型性能降低
+export SCALAR_TO_HOST_MEM=1
 
 ulimit -SHn 512000
 
-- 
Gitee