From d2432a962ab798b18b4de08aff97df619de1b008 Mon Sep 17 00:00:00 2001
From: rainyhorse <zhengbin20@huawei.com>
Date: Tue, 9 Mar 2021 10:36:25 +0800
Subject: [PATCH] update the convert tool path

---
 tests/st/dump/test_data_dump.py | 37 ++++++++++++++++++++++++---------
 1 file changed, 27 insertions(+), 10 deletions(-)

diff --git a/tests/st/dump/test_data_dump.py b/tests/st/dump/test_data_dump.py
index 8e71e22931..bea56efe7c 100644
--- a/tests/st/dump/test_data_dump.py
+++ b/tests/st/dump/test_data_dump.py
@@ -102,6 +102,20 @@ class ReluReduceMeanDenseRelu(Cell):
         x_ = self.relu(x_)
         return x_
 
+
+def search_path(path, keyword):
+    content = os.listdir(path)
+    for each in content:
+        each_path = path + os.sep + each
+        if keyword in each:
+            return each_path
+        read_write = os.access(each_path, os.W_OK) and os.access(each_path, os.R_OK)
+        if not read_write:
+            continue
+        if os.path.isdir(each_path):
+            search_path(each_path, keyword)
+    return None
+
 @pytest.mark.level0
 @pytest.mark.platform_arm_ascend_training
 @pytest.mark.platform_x86_ascend_training
@@ -136,13 +150,16 @@ def test_async_dump_net_multi_layer_mode1():
     if os.path.exists(npy_path):
         shutil.rmtree(npy_path)
     os.mkdir(npy_path)
-    cmd = "python /usr/local/Ascend/toolkit/tools/operator_cmp/compare/msaccucmp.pyc " \
-        "convert -d {0} -out {1}".format(dump_file_full_path, npy_path)
-    os.system(cmd)
-    npy_file_list = os.listdir(npy_path)
-    dump_result = {}
-    for file in npy_file_list:
-        if "output.0.npy" in file:
-            dump_result["output0"] = np.load(os.path.join(npy_path, file))
-    for index, value in enumerate(net_dict):
-        assert value.asnumpy() == dump_result["output0"][index]
+    tool_path = search_path('/usr/local/Ascend', 'msaccucmp.pyc')
+    if tool_path:
+        cmd = "python {0} convert -d {1} -out {2}".format(tool_path, dump_file_full_path, npy_path)
+        os.system(cmd)
+        npy_file_list = os.listdir(npy_path)
+        dump_result = {}
+        for file in npy_file_list:
+            if "output.0.npy" in file:
+                dump_result["output0"] = np.load(os.path.join(npy_path, file))
+        for index, value in enumerate(net_dict):
+            assert value.asnumpy() == dump_result["output0"][index]
+    else:
+        print('not find convert tools msaccucmp.pyc')