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

!48399 Remove ST from GRUV2 and LSTMV2 

Merge pull request !48399 from liuluobin/master_remove_rnnv2_st
This commit is contained in:
i-robot 2023-02-04 03:33:08 +00:00 committed by Gitee
parent a4b2afc0ea 5bc1b091df
commit a3ec17fc5a
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/st/ops/gpu/test_gruv2_op.py
View File

@ -60,7 +60,6 @@ def get_weights_from_gru(gru_nn, has_bias):
return weights
@pytest.mark.level1
@pytest.mark.platform_x86_gpu_training
@pytest.mark.env_onecard
@pytest.mark.parametrize("has_bias", [True, False])
@ -94,7 +93,6 @@ def test_gruv2_op(has_bias, is_train, dtype):
assert np.allclose(me_hy.asnumpy(), expect_hy.asnumpy(), rtol, atol)
@pytest.mark.level0
@pytest.mark.platform_x86_gpu_training
@pytest.mark.env_onecard
def test_gruv2_op_float64_exception():

2
tests/st/ops/gpu/test_lstmv2_op.py
View File

@ -60,7 +60,6 @@ def get_weights_from_lstm(lstm_nn, has_bias):
return weights
@pytest.mark.level1
@pytest.mark.platform_x86_gpu_training
@pytest.mark.env_onecard
@pytest.mark.parametrize("has_bias", [True, False])
@ -97,7 +96,6 @@ def test_lstmv2_op(has_bias, is_train, dtype):
assert np.allclose(me_cy.asnumpy(), expect_cy.asnumpy(), rtol, atol)
@pytest.mark.level0
@pytest.mark.platform_x86_gpu_training
@pytest.mark.env_onecard
def test_lstmv2_op_float64_exception():