|
|
|
@@ -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') |