Browse Source

update graphengine 0311

tags/v1.2.0-rc1
shenwei41 4 years ago
parent
commit
c0f966ddef
7 changed files with 8 additions and 7 deletions
  1. +0
    -1
      cmake/dependency_graphengine.cmake
  2. +1
    -1
      graphengine
  3. +1
    -1
      mindspore/_check_version.py
  4. +3
    -1
      mindspore/_extends/parallel_compile/tbe_compiler/compiler.py
  5. +1
    -1
      tests/st/dynamic_shape/test_ascend_cpu.py
  6. +1
    -1
      tests/st/dynamic_shape/test_dynamic_shape_embedding.py
  7. +1
    -1
      tests/st/dynamic_shape/test_ftrl.py

+ 0
- 1
cmake/dependency_graphengine.cmake 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
graphengine

@@ -1 +1 @@
Subproject commit 92286b21ec71014f547715067d15edb9c2e26a38
Subproject commit f65be61197ed36dfc9dc10b91b58bf93835fa27b

+ 1
- 1
mindspore/_check_version.py 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"


+ 3
- 1
mindspore/_extends/parallel_compile/tbe_compiler/compiler.py 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


+ 1
- 1
tests/st/dynamic_shape/test_ascend_cpu.py 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


+ 1
- 1
tests/st/dynamic_shape/test_dynamic_shape_embedding.py 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


+ 1
- 1
tests/st/dynamic_shape/test_ftrl.py 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


Loading…
Cancel
Save