From b349745a4373e96f6501974981f2c5bedc259c3d Mon Sep 17 00:00:00 2001
From: litian_drinksnow <litian68@huawei.com>
Date: Thu, 3 Aug 2023 20:47:01 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A4=9A=E5=8D=A1=E9=80=82=E9=85=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 debug/accuracy_tools/api_accuracy_checker/dump/api_info.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/debug/accuracy_tools/api_accuracy_checker/dump/api_info.py b/debug/accuracy_tools/api_accuracy_checker/dump/api_info.py
index e8c085a0b88..0e9edec8a6d 100644
--- a/debug/accuracy_tools/api_accuracy_checker/dump/api_info.py
+++ b/debug/accuracy_tools/api_accuracy_checker/dump/api_info.py
@@ -1,4 +1,5 @@
 # 定义API INFO，保存基本信息，用于后续结构体的落盘，注意考虑random场景及真实数据场景
+import os
 import inspect
 import torch
 import torch_npu
@@ -9,7 +10,7 @@ from api_accuracy_checker.dump.utils import write_npy
 
 class APIInfo:
     def __init__(self, api_name):
-        self.rank = torch_npu.npu.current_device()
+        self.rank = os.getpid()
         self.api_name = api_name
         self.save_real_data = msCheckerConfig.real_data
 
-- 
Gitee