!31779 fix cmake script and adapt to new file stucture

Merge pull request !31779 from yanghaoran/master
This commit is contained in:
i-robot 2022-03-23 09:10:18 +00:00 committed by Gitee
commit e36fdde9ec
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 2 additions and 2 deletions

View File

@ -34,7 +34,7 @@ else()
set(ASCEND_TOOLKIT_RUNTIME_PATH ${ASCEND_PATH}/ascend-toolkit/latest/fwkacllib/lib64)
set(ASCEND_TOOLKIT_PLUGIN_PATH ${ASCEND_TOOLKIT_RUNTIME_PATH}/plugin/opskernel)
set(ASCEND_CANN_RUNTIME_PATH ${ASCEND_PATH}/latest/fwkacllib/lib64)
set(ASCEND_CANN_PLUGIN_PATH ${ASCEND_RUNTIME_PATH}/latest/plugin/opskernel)
set(ASCEND_CANN_PLUGIN_PATH ${ASCEND_CANN_RUNTIME_PATH}/plugin/opskernel)
endif()
find_library(HCCL hccl ${ASCEND_RUNTIME_PATH} ${ASCEND_TOOLKIT_RUNTIME_PATH} ${ASCEND_CANN_RUNTIME_PATH})

View File

@ -131,7 +131,7 @@ def test_sit_lstm_forward_input_3_32_32_is_32_hs_16():
assert np.allclose(hy.asnumpy(), hy_pynative.asnumpy(), 0.002, 0.002)
assert np.allclose(cy.asnumpy(), cy_pynative.asnumpy(), 0.002, 0.002)
@pytest.mark.level0
@pytest.mark.level1
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training
@pytest.mark.env_onecard