update graphengine 0311

This commit is contained in:
shenwei41 2021-03-12 15:07:38 +08:00
parent 54bbc5e545
commit c0f966ddef
7 changed files with 8 additions and 7 deletions

View File

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

View File

@ -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"

View File

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

View File

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

View File

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

View File

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