diff --git a/debug/accuracy_tools/api_accuracy_checker/compare/compare.py b/debug/accuracy_tools/api_accuracy_checker/compare/compare.py
index 3c9b5e9214738162344fa60e400178b0953d4aa7..30bf805e716350c381d0e8358473509e66308db5 100644
--- a/debug/accuracy_tools/api_accuracy_checker/compare/compare.py
+++ b/debug/accuracy_tools/api_accuracy_checker/compare/compare.py
@@ -182,8 +182,8 @@ class Comparator:
 
     def compare_output(self, full_api_name, data_info):
         _, api_name, _ = full_api_name.split("*")
-        bench_output, device_output = data_info.bench_out, data_info.device_out
-        bench_grad, device_grad = data_info.bench_grad_out, data_info.device_grad_out
+        bench_output, device_output = data_info.bench_output, data_info.device_output
+        bench_grad, device_grad = data_info.bench_grad, data_info.device_grad
         backward_message = data_info.backward_message
         compare_func = self._compare_dropout if "dropout" in full_api_name else self._compare_core_wrapper
         # forward result compare
diff --git a/debug/accuracy_tools/api_accuracy_checker/run_ut/run_ut.py b/debug/accuracy_tools/api_accuracy_checker/run_ut/run_ut.py
index 3aacb9e17cfb8bd79f1b3961f29e465d39c164b3..a3353cc75d1f7f9b74a95ff2f6ddd0ee44008fad 100644
--- a/debug/accuracy_tools/api_accuracy_checker/run_ut/run_ut.py
+++ b/debug/accuracy_tools/api_accuracy_checker/run_ut/run_ut.py
@@ -538,12 +538,12 @@ def run_ut_command(args):
 
 
 class UtDataInfo:
-    def __init__(self, bench_grad_out, device_grad_out, device_out,
-                 bench_out, grad_in, in_fwd_data_list, backward_message, rank=0):
-        self.bench_grad_out = bench_grad_out
-        self.device_grad_out = device_grad_out
-        self.device_out = device_out
-        self.bench_out = bench_out
+    def __init__(self, bench_grad, device_grad, device_output, bench_output, grad_in, in_fwd_data_list, 
+                 backward_message, rank=0):
+        self.bench_grad = bench_grad
+        self.device_grad = device_grad
+        self.device_output = device_output
+        self.bench_output = bench_output
         self.grad_in = grad_in
         self.in_fwd_data_list = in_fwd_data_list
         self.backward_message = backward_message
diff --git a/debug/accuracy_tools/api_accuracy_checker/test/ut/run_ut/test_run_ut.py b/debug/accuracy_tools/api_accuracy_checker/test/ut/run_ut/test_run_ut.py
index 3412620f7164a017181d7872824796c147a8ca72..3c180fa23b3e27bdd2ccd91698320684f007e47a 100644
--- a/debug/accuracy_tools/api_accuracy_checker/test/ut/run_ut/test_run_ut.py
+++ b/debug/accuracy_tools/api_accuracy_checker/test/ut/run_ut/test_run_ut.py
@@ -62,10 +62,10 @@ class TestRunUtMethods(unittest.TestCase):
     
     def test_UtDataInfo(self):
         data_info = UtDataInfo(None, None, None, None, None, None, None)
-        self.assertIsNone(data_info.bench_grad_out)
-        self.assertIsNone(data_info.device_grad_out)
-        self.assertIsNone(data_info.device_out)
-        self.assertIsNone(data_info.bench_out)
+        self.assertIsNone(data_info.bench_grad)
+        self.assertIsNone(data_info.device_grad)
+        self.assertIsNone(data_info.device_output)
+        self.assertIsNone(data_info.bench_output)
         self.assertIsNone(data_info.grad_in)
         self.assertIsNone(data_info.in_fwd_data_list)
         self.assertIsNone(data_info.backward_message)