add @security_off_wrap for RDR testcases

This commit is contained in:
liangyongxiong 2021-12-23 10:37:05 +08:00
parent 03f969c377
commit 58ff7b1e07
4 changed files with 10 additions and 0 deletions

View File

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

View File

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

View File

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

View File

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