forked from mindspore-Ecosystem/mindspore
add @security_off_wrap for RDR testcases
This commit is contained in:
parent
03f969c377
commit
58ff7b1e07
|
@ -18,6 +18,7 @@ import json
|
|||
import pytest
|
||||
|
||||
import mindspore.context as context
|
||||
from tests.security_utils import security_off_wrap
|
||||
from .test_tdt_data_transfer import test_tdt_consume_beyond_produce
|
||||
|
||||
# create config file for RDR
|
||||
|
@ -35,12 +36,14 @@ def test_train(device_type):
|
|||
context.set_context(env_config_path=config_file)
|
||||
test_tdt_consume_beyond_produce()
|
||||
|
||||
@security_off_wrap
|
||||
@pytest.mark.level0
|
||||
@pytest.mark.platform_arm_ascend_training
|
||||
@pytest.mark.env_onecard
|
||||
def test_train_with_Ascend():
|
||||
test_train("Ascend")
|
||||
|
||||
@security_off_wrap
|
||||
@pytest.mark.level0
|
||||
@pytest.mark.platform_x86_gpu_training
|
||||
@pytest.mark.env_onecard
|
||||
|
|
|
@ -4,6 +4,7 @@ import json
|
|||
import pytest
|
||||
|
||||
import mindspore.context as context
|
||||
from tests.security_utils import security_off_wrap
|
||||
from .test_network_main import test_lenet
|
||||
|
||||
# create config file for RDR
|
||||
|
@ -21,6 +22,7 @@ def test_train(device_type):
|
|||
context.set_context(env_config_path=config_file)
|
||||
test_lenet()
|
||||
|
||||
@security_off_wrap
|
||||
@pytest.mark.level0
|
||||
@pytest.mark.platform_arm_ascend_training
|
||||
@pytest.mark.env_onecard
|
||||
|
|
|
@ -4,6 +4,7 @@ import json
|
|||
import pytest
|
||||
|
||||
import mindspore.context as context
|
||||
from tests.security_utils import security_off_wrap
|
||||
from .test_network_main import test_lenet
|
||||
|
||||
# create config file for RDR
|
||||
|
@ -21,6 +22,7 @@ def test_train(device_type):
|
|||
context.set_context(env_config_path=config_file)
|
||||
test_lenet()
|
||||
|
||||
@security_off_wrap
|
||||
@pytest.mark.level0
|
||||
@pytest.mark.platform_x86_cpu
|
||||
@pytest.mark.env_onecard
|
||||
|
|
|
@ -4,6 +4,7 @@ import json
|
|||
import pytest
|
||||
|
||||
import mindspore.context as context
|
||||
from tests.security_utils import security_off_wrap
|
||||
from .test_gpu_lenet import test_train_lenet, test_train_and_eval_lenet
|
||||
|
||||
# create config file for RDR
|
||||
|
@ -29,12 +30,14 @@ def test_train_and_eval():
|
|||
# train and eval with GPU
|
||||
test_train_and_eval_lenet()
|
||||
|
||||
@security_off_wrap
|
||||
@pytest.mark.level0
|
||||
@pytest.mark.platform_x86_gpu_training
|
||||
@pytest.mark.env_onecard
|
||||
def test_train_with_GPU():
|
||||
test_train("GPU")
|
||||
|
||||
@security_off_wrap
|
||||
@pytest.mark.level0
|
||||
@pytest.mark.platform_x86_gpu_training
|
||||
@pytest.mark.env_onecard
|
||||
|
|
Loading…
Reference in New Issue