From da15d38c0b1b7c0d99111e5930acc1c71424208b Mon Sep 17 00:00:00 2001
From: cai-weiwei1989 <734267852@qq.com>
Date: Mon, 25 Mar 2024 11:39:17 +0800
Subject: [PATCH] =?UTF-8?q?[compare=5Ftools]=E4=BD=8E=E9=94=99=E4=BF=AE?=
 =?UTF-8?q?=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 profiler/compare_tools/README.md | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/profiler/compare_tools/README.md b/profiler/compare_tools/README.md
index 106f53e11..014fa36e4 100644
--- a/profiler/compare_tools/README.md
+++ b/profiler/compare_tools/README.md
@@ -32,11 +32,11 @@ pip3 install numpy
 采集样例代码参考一：
 
 ```Python
-with torch.profiler.profile(
+with torch_npu.profiler.profile(
         profile_memory=True,  # 内存数据采集的开关
         record_shapes=True,  # 算子input shape信息采集的开关
-        schedule=torch.profiler.schedule(wait=10, warmup=0, active=1, repeat=1),
-        on_trace_ready=torch.profiler.tensorboard_trace_handler("./result_dir")
+        schedule=torch_npu.profiler.schedule(wait=10, warmup=0, active=1, repeat=1),
+        on_trace_ready=torch_npu.profiler.tensorboard_trace_handler("./result_dir")
 ) as prof:
     for step in ranges(step_number):
         train_one_step()
@@ -46,10 +46,10 @@ with torch.profiler.profile(
 采集样例代码参考二：
 
 ```Python
-prof = torch.profiler.profile(
+prof = torch_npu.profiler.profile(
     profile_memory=True,  # 内存数据采集的开关
     record_shapes=True,  # 算子input shape信息采集的开关
-    on_trace_ready=torch.profiler.tensorboard_trace_handler("./result_dir"))
+    on_trace_ready=torch_npu.profiler.tensorboard_trace_handler("./result_dir"))
 for step in range(step_number):
     if step == 11:
         prof.start()
-- 
Gitee