From c83add51a6ad62d2f791755ff69c644b53b1d415 Mon Sep 17 00:00:00 2001
From: Qiyu8 <fangchunlin@huawei.com>
Date: Thu, 31 Mar 2022 14:08:38 +0800
Subject: [PATCH] add gpu env setting

---
 src/analysis.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/analysis.py b/src/analysis.py
index d8a3bf6..fad7b53 100644
--- a/src/analysis.py
+++ b/src/analysis.py
@@ -473,6 +473,7 @@ perf report  -i ./perf.data -F period,sample,overhead,symbol,dso,comm -s overhea
         print(f"start gpu perf")
         run_cmd = self.hpc_data.get_run()
         gperf_cmd = f'''
+{self.hpc_data.get_env()}
 cd {Data.case_dir}
 nsys profile -y 5s -d 100s {Data.nsys_para} -o nsys-{self.get_arch()}-{self.get_cur_time()} {run_cmd}
     '''
@@ -482,9 +483,10 @@ nsys profile -y 5s -d 100s {Data.nsys_para} -o nsys-{self.get_arch()}-{self.get_
         print(f"start ncu perf")
         run_cmd = self.hpc_data.get_run()
         ncu_cmd = f'''
-    cd {Data.case_dir}
-    ncu --export ncu-{self.get_arch()}-{self.get_cur_time()} {Data.ncu_para} --import-source=yes --set full --kernel-name {kernel} --launch-skip 1735 --launch-count 1 {run_cmd}
-    '''
+{self.hpc_data.get_env()}
+cd {Data.case_dir}
+ncu --export ncu-{self.get_arch()}-{self.get_cur_time()} {Data.ncu_para} --import-source=yes --set full --kernel-name {kernel} --launch-skip 1735 --launch-count 1 {run_cmd}
+'''
         self.exe.exec_raw(ncu_cmd)
 
 class Download:
-- 
Gitee