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 e8c085a0b885fc2e8b71d1f88e7ae6a68faebd6b..0e9edec8a6d8a2a3e661354e428324f7b3cd7330 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