From 94f2fa0273b62b8598ac2293963b91e56067fca3 Mon Sep 17 00:00:00 2001
From: jiangchao_j <chaojiang_j@163.com>
Date: Mon, 30 Jun 2025 20:38:26 +0800
Subject: [PATCH] determine whether it is MSAdapter

---
 debug/accuracy_tools/msprobe/docs/29.data_dump_MSAdapter.md | 2 ++
 debug/accuracy_tools/msprobe/mindspore/common/utils.py      | 3 +++
 2 files changed, 5 insertions(+)

diff --git a/debug/accuracy_tools/msprobe/docs/29.data_dump_MSAdapter.md b/debug/accuracy_tools/msprobe/docs/29.data_dump_MSAdapter.md
index 6549b15e7a..a290d349de 100644
--- a/debug/accuracy_tools/msprobe/docs/29.data_dump_MSAdapter.md
+++ b/debug/accuracy_tools/msprobe/docs/29.data_dump_MSAdapter.md
@@ -4,6 +4,8 @@ MSAdapter 是一款 MindSpore 生态适配工具，可以将 PyTorch 训练脚
 
 msprobe 工具主要通过在训练脚本内添加 dump 接口、启动训练的方式采集精度数据。
 
+**注意**：为了正确识别 MSAdapter 场景，在导入 msprobe 工具前，需完成 torch 模块的的导入。
+
 本工具提供固定的 API 支持列表，若需要删除或增加 dump 的 API，可以在 msprobe/pytorch/hook_module/support_wrap_ops.yaml 文件内手动修改，如下示例：
 
 ```yaml
diff --git a/debug/accuracy_tools/msprobe/mindspore/common/utils.py b/debug/accuracy_tools/msprobe/mindspore/common/utils.py
index 72e09c7609..e4646e906d 100644
--- a/debug/accuracy_tools/msprobe/mindspore/common/utils.py
+++ b/debug/accuracy_tools/msprobe/mindspore/common/utils.py
@@ -16,6 +16,7 @@
 import inspect
 import os
 import random
+import sys
 import types
 
 import mindspore as ms
@@ -176,6 +177,8 @@ def is_mindtorch():
     global mindtorch_check_result
     if mindtorch_check_result is None:
         mindtorch_check_result = False
+        if 'torch' not in sys.modules:
+            return mindtorch_check_result
         try:
             import torch
         except ImportError:
-- 
Gitee