forked from mindspore-Ecosystem/mindspore
update graphengine 0311
This commit is contained in:
parent
54bbc5e545
commit
c0f966ddef
|
@ -30,7 +30,6 @@ if(ENABLE_D OR ENABLE_ACL OR ENABLE_TESTCASES)
|
|||
set(GE_PREBUILD_PATH ${GE_SOURCE_DIR}/third_party/prebuild/${CMAKE_HOST_SYSTEM_PROCESSOR})
|
||||
set(ENABLE_MS_TESTCASES TRUE)
|
||||
find_submodule_lib(slog libalog.so ${GE_PREBUILD_PATH})
|
||||
find_submodule_lib(error_manager liberror_manager.so ${GE_PREBUILD_PATH})
|
||||
find_submodule_lib(static_mmpa libmmpa.a ${GE_PREBUILD_PATH})
|
||||
endif()
|
||||
|
||||
|
|
|
@ -1 +1 @@
|
|||
Subproject commit 92286b21ec71014f547715067d15edb9c2e26a38
|
||||
Subproject commit f65be61197ed36dfc9dc10b91b58bf93835fa27b
|
|
@ -166,7 +166,7 @@ class AscendEnvChecker(EnvChecker):
|
|||
"""ascend environment check"""
|
||||
|
||||
def __init__(self):
|
||||
self.version = ["1.77.T20.0.B200"]
|
||||
self.version = ["1.77.22.0.220"]
|
||||
atlas_nnae_version = "/usr/local/Ascend/nnae/latest/fwkacllib/version.info"
|
||||
atlas_toolkit_version = "/usr/local/Ascend/ascend-toolkit/latest/fwkacllib/version.info"
|
||||
hisi_fwk_version = "/usr/local/Ascend/fwkacllib/version.info"
|
||||
|
|
|
@ -114,7 +114,9 @@ def build_op(build_type, json_str, tune_mode=None):
|
|||
|
||||
# call function
|
||||
if is_dynamic_shape:
|
||||
with te.op.dynamic():
|
||||
# with te.op.dynamic():
|
||||
import tbe.common.context.op_context as op_context
|
||||
with op_context.OpContext("dynamic"):
|
||||
op_func(*inputs_args, *outputs_args, *attrs_args, kernel_name=kernel_name)
|
||||
if tune_mode is not None:
|
||||
return (te.op.get_compile_info()), (inputs_args, outputs_args, attrs_args), op_module_name
|
||||
|
|
|
@ -58,7 +58,7 @@ def test_unique_ascend():
|
|||
assert (output[1].asnumpy() == expect2).all()
|
||||
|
||||
|
||||
@pytest.mark.level0
|
||||
@pytest.mark.level2
|
||||
@pytest.mark.platform_arm_ascend_training
|
||||
@pytest.mark.platform_x86_ascend_training
|
||||
@pytest.mark.env_onecard
|
||||
|
|
|
@ -36,7 +36,7 @@ class NetWithEmbeddingLookUp(nn.Cell):
|
|||
return out
|
||||
|
||||
|
||||
@pytest.mark.level0
|
||||
@pytest.mark.level2
|
||||
@pytest.mark.platform_arm_ascend_training
|
||||
@pytest.mark.platform_x86_ascend_training
|
||||
@pytest.mark.platform_x86_gpu_training
|
||||
|
|
|
@ -56,7 +56,7 @@ def test_ftrl_net():
|
|||
[[0.6821311, 0.6821311]],
|
||||
[[0.6821311, 0.6821311]]]))
|
||||
|
||||
@pytest.mark.level0
|
||||
@pytest.mark.level2
|
||||
@pytest.mark.platform_arm_ascend_training
|
||||
@pytest.mark.platform_x86_ascend_training
|
||||
@pytest.mark.env_onecard
|
||||
|
|
Loading…
Reference in New Issue