Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!35111 fix op fusion 

Merge pull request !35111 from liubuyu/usb
This commit is contained in:
i-robot 2022-05-30 01:07:33 +00:00 committed by Gitee
parent 6627f5aa33 8aabd90519
commit 359551ba3f
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/python/mindspore/_extends/parallel_compile/tbe_compiler/tbe_adapter.py
View File

@ -522,8 +522,9 @@ def parallel_compile_fusion_op(job: TbeJob):
options = get_options_info(job.content)
op_kernel_name = job.content["fusion_op_name"]
op_name = job.content["full_name"]
relation = ""
dispatch_fusion_op_compile_task(job.source_id, job.id, l1_size, json.dumps(job.content), op_kernel_name, None, None,
options, None, job.pass_list, op_name)
options, None, job.pass_list, op_name, relation)
return True

20
tests/st/numpy_native/test_math_ops.py
View File

@ -469,7 +469,7 @@ def onp_average(x):
return a, b, c, d, e, f, g, h, i
@pytest.mark.level1
@pytest.mark.level0
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training
@pytest.mark.platform_x86_gpu_training
@ -1283,7 +1283,7 @@ def onp_kron(x, y):
return onp.kron(x, y)
@pytest.mark.level1
@pytest.mark.level0
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training
@pytest.mark.platform_x86_gpu_training
@ -1303,7 +1303,7 @@ def test_kron():
match_res(mnp.kron, onp.kron, x, y)
@pytest.mark.level1
@pytest.mark.level0
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training
@pytest.mark.platform_x86_gpu_training
@ -1437,7 +1437,7 @@ def onp_diff(input_array):
return a, b, c, d, e, f, g
@pytest.mark.level1
@pytest.mark.level0
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training
@pytest.mark.platform_x86_gpu_training
@ -1743,7 +1743,7 @@ def test_convolve():
match_all_arrays(mnp_res, onp_res)
@pytest.mark.level1
@pytest.mark.level0
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training
@pytest.mark.platform_x86_gpu_training
@ -2003,7 +2003,7 @@ def test_mean():
run_multi_test(mnp_mean, onp_mean, test_case.scalars, error=3)
@pytest.mark.level1
@pytest.mark.level0
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training
@pytest.mark.platform_x86_gpu_training
@ -2034,7 +2034,7 @@ def test_multi_dot():
match_all_arrays(mnp.multi_dot(mnp_arrays[1:-1]), onp.linalg.multi_dot(arrays[1:-1]))
@pytest.mark.level1
@pytest.mark.level0
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training
@pytest.mark.platform_x86_gpu_training
@ -2148,7 +2148,7 @@ def test_bincount():
onp.bincount(x, weights, minlength=25), error=3)
@pytest.mark.level1
@pytest.mark.level0
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training
@pytest.mark.platform_x86_gpu_training
@ -2472,7 +2472,7 @@ def test_result_type():
assert actual == expected
@pytest.mark.level1
@pytest.mark.level0
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training
@pytest.mark.platform_x86_gpu_training
@ -2555,7 +2555,7 @@ def test_ravel_multi_index():
match_array(actual.asnumpy(), expected, error=5)
@pytest.mark.level1
@pytest.mark.level0
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training
@pytest.mark.platform_x86_gpu_training