diff --git a/OWNERS b/OWNERS index e0a0b376144..d57557064fa 100644 --- a/OWNERS +++ b/OWNERS @@ -19,6 +19,9 @@ files: - guoqi1024 - baochong - liujunzhu + - kisnwang + - yuchaojie + - hwcaifubi "version.txt": approvers: diff --git a/tests/st/ge/test_ge_jit_level.py b/tests/st/ge/test_ge_jit_level.py index 3df3d19bc99..f605bfe4d39 100644 --- a/tests/st/ge/test_ge_jit_level.py +++ b/tests/st/ge/test_ge_jit_level.py @@ -31,6 +31,7 @@ def run_testcase(file_name, case_name=""): @pytest.mark.level0 @pytest.mark.platform_arm_ascend_training @pytest.mark.platform_x86_ascend_training +@pytest.mark.platform_arm_ascend910b_training @pytest.mark.env_onecard def test_ge_graph_mode_with_jit_level(): """ diff --git a/tests/st/ge/test_if_by_if.py b/tests/st/ge/test_if_by_if.py index 5f7d7dca85b..dda3b3ce7e3 100644 --- a/tests/st/ge/test_if_by_if.py +++ b/tests/st/ge/test_if_by_if.py @@ -19,6 +19,7 @@ import pytest @pytest.mark.level0 @pytest.mark.platform_x86_ascend_training @pytest.mark.platform_arm_ascend_training +@pytest.mark.platform_arm_ascend910b_training @pytest.mark.env_onecard def test_if_by_if(): """ diff --git a/tests/st/ge/test_sparse_tensor.py b/tests/st/ge/test_sparse_tensor.py index d3388ea8ae2..f31f60cb4ba 100644 --- a/tests/st/ge/test_sparse_tensor.py +++ b/tests/st/ge/test_sparse_tensor.py @@ -19,6 +19,7 @@ import tests.st.ge.ge_test_utils as utils @pytest.mark.level0 @pytest.mark.platform_arm_ascend_training @pytest.mark.platform_x86_ascend_training +@pytest.mark.platform_arm_ascend910b_training @pytest.mark.env_onecard def test_sparse_tensor(): """ diff --git a/tests/st/ge/test_while_by_cell_list_in_while.py b/tests/st/ge/test_while_by_cell_list_in_while.py index c926190215a..715b15b917a 100644 --- a/tests/st/ge/test_while_by_cell_list_in_while.py +++ b/tests/st/ge/test_while_by_cell_list_in_while.py @@ -19,6 +19,7 @@ import pytest @pytest.mark.level0 @pytest.mark.platform_x86_ascend_training @pytest.mark.platform_arm_ascend_training +@pytest.mark.platform_arm_ascend910b_training @pytest.mark.env_onecard def test_while_by_cell_list_in_while(): """ diff --git a/tests/st/model_zoo_tests/DeepFM/test_deepfm.py b/tests/st/model_zoo_tests/DeepFM/test_deepfm.py index 356ed82c87d..9031ef641bb 100644 --- a/tests/st/model_zoo_tests/DeepFM/test_deepfm.py +++ b/tests/st/model_zoo_tests/DeepFM/test_deepfm.py @@ -28,8 +28,9 @@ from src.callback import EvalCallBack, LossCallBack, TimeMonitor set_seed(1) -@pytest.mark.level1 +@pytest.mark.level0 @pytest.mark.platform_arm_ascend_training +@pytest.mark.platform_arm_ascend910b_training @pytest.mark.platform_x86_ascend_training @pytest.mark.env_onecard def test_deepfm(): diff --git a/tests/st/model_zoo_tests/deeplabv3/test_DeeplabV3_voc2007.py b/tests/st/model_zoo_tests/deeplabv3/test_DeeplabV3_voc2007.py index 6d01dba4089..3f5268f63f0 100644 --- a/tests/st/model_zoo_tests/deeplabv3/test_DeeplabV3_voc2007.py +++ b/tests/st/model_zoo_tests/deeplabv3/test_DeeplabV3_voc2007.py @@ -18,9 +18,10 @@ import pytest from tests.st.model_zoo_tests import utils -@pytest.mark.level1 +@pytest.mark.level0 @pytest.mark.platform_x86_ascend_training @pytest.mark.platform_arm_ascend_training +@pytest.mark.platform_arm_ascend910b_training @pytest.mark.env_single def test_DeeplabV3_voc2007(): cur_path = os.path.dirname(os.path.abspath(__file__)) diff --git a/tests/st/model_zoo_tests/resnet50/test_resnet50_imagenet2012.py b/tests/st/model_zoo_tests/resnet50/test_resnet50_imagenet2012.py index 468c2da771d..9c35b025dd5 100644 --- a/tests/st/model_zoo_tests/resnet50/test_resnet50_imagenet2012.py +++ b/tests/st/model_zoo_tests/resnet50/test_resnet50_imagenet2012.py @@ -26,6 +26,7 @@ np.random.seed(1) @pytest.mark.level0 @pytest.mark.platform_x86_ascend_training @pytest.mark.platform_arm_ascend_training +@pytest.mark.platform_arm_ascend910b_training @pytest.mark.env_single def test_ge_resnet50_boost_imagenet2012_ascend(): """ diff --git a/tests/st/model_zoo_tests/ssd_mobilenet_fpn/test_ssd_modelnet_fpn.py b/tests/st/model_zoo_tests/ssd_mobilenet_fpn/test_ssd_modelnet_fpn.py index 47e5fbc8adc..b1bdfd3ab42 100644 --- a/tests/st/model_zoo_tests/ssd_mobilenet_fpn/test_ssd_modelnet_fpn.py +++ b/tests/st/model_zoo_tests/ssd_mobilenet_fpn/test_ssd_modelnet_fpn.py @@ -19,9 +19,10 @@ import pytest from tests.st.model_zoo_tests import utils -@pytest.mark.level1 +@pytest.mark.level0 @pytest.mark.platform_x86_ascend_training @pytest.mark.platform_arm_ascend_training +@pytest.mark.platform_arm_ascend910b_training @pytest.mark.env_single def test_SSD_mobilenet_v1_fpn_coco2017(): cur_path = os.path.dirname(os.path.abspath(__file__)) diff --git a/tests/st/model_zoo_tests/wide_and_deep/test_wide_and_deep_auto_parallel.py b/tests/st/model_zoo_tests/wide_and_deep/test_wide_and_deep_auto_parallel.py index 10085c0b123..f6ef2b20cc6 100644 --- a/tests/st/model_zoo_tests/wide_and_deep/test_wide_and_deep_auto_parallel.py +++ b/tests/st/model_zoo_tests/wide_and_deep/test_wide_and_deep_auto_parallel.py @@ -16,9 +16,10 @@ import os import pytest -@pytest.mark.level1 +@pytest.mark.level0 @pytest.mark.platform_x86_ascend_training @pytest.mark.platform_arm_ascend_training +@pytest.mark.platform_arm_ascend910b_training @pytest.mark.env_single def test_wide_and_deep(): sh_path = os.path.split(os.path.realpath(__file__))[0] diff --git a/tests/st/model_zoo_tests/yolov3_darknet53/test_yolov3_darknet53.py b/tests/st/model_zoo_tests/yolov3_darknet53/test_yolov3_darknet53.py index c8be636c6cc..d1a1d8885d9 100644 --- a/tests/st/model_zoo_tests/yolov3_darknet53/test_yolov3_darknet53.py +++ b/tests/st/model_zoo_tests/yolov3_darknet53/test_yolov3_darknet53.py @@ -91,9 +91,10 @@ class TimeMonitor(Callback): DATA_DIR = "/home/workspace/mindspore_dataset/coco/coco2014/" -@pytest.mark.level1 +@pytest.mark.level0 @pytest.mark.platform_arm_ascend_training @pytest.mark.platform_x86_ascend_training +@pytest.mark.platform_arm_ascend910b_training @pytest.mark.env_onecard def test_yolov3_darknet53(): devid = int(os.getenv('DEVICE_ID')) if os.getenv('DEVICE_ID') else 0 @@ -213,9 +214,10 @@ def test_yolov3_darknet53(): print('==========test case passed===========') -@pytest.mark.level1 +@pytest.mark.level0 @pytest.mark.platform_x86_ascend_training @pytest.mark.platform_arm_ascend_training +@pytest.mark.platform_arm_ascend910b_training @pytest.mark.env_single def test_yolov3_darknet_8p(): cur_path = os.path.dirname(os.path.abspath(__file__)) diff --git a/tests/st/networks/models/bert/bert_performance/test_bert_tdt_ge_run.py b/tests/st/networks/models/bert/bert_performance/test_bert_tdt_ge_run.py index 4970af7d533..3a11ce869f5 100644 --- a/tests/st/networks/models/bert/bert_performance/test_bert_tdt_ge_run.py +++ b/tests/st/networks/models/bert/bert_performance/test_bert_tdt_ge_run.py @@ -22,6 +22,7 @@ import numpy as np @pytest.mark.level0 @pytest.mark.platform_arm_ascend_training @pytest.mark.platform_x86_ascend_training +@pytest.mark.platform_arm_ascend910b_training @pytest.mark.env_onecard def test_bert_for_ge(): """" diff --git a/tests/st/networks/test_lenet.py b/tests/st/networks/test_lenet.py index e62b67984cf..05e80de177a 100644 --- a/tests/st/networks/test_lenet.py +++ b/tests/st/networks/test_lenet.py @@ -356,6 +356,7 @@ class CustomCallback(Callback): @pytest.mark.level0 @pytest.mark.platform_arm_ascend_training +@pytest.mark.platform_arm_ascend910b_training @pytest.mark.env_onecard def test_pynative_tasksink(): """ diff --git a/tests/st/pynative/test_tensor_setitem.py b/tests/st/pynative/test_tensor_setitem.py index 01f721a795a..93b0fd396ad 100644 --- a/tests/st/pynative/test_tensor_setitem.py +++ b/tests/st/pynative/test_tensor_setitem.py @@ -240,6 +240,7 @@ class TensorItemSetWithNumber(Cell): @pytest.mark.level0 @pytest.mark.platform_arm_ascend_training @pytest.mark.platform_x86_ascend_training +@pytest.mark.platform_arm_ascend910b_training @pytest.mark.platform_x86_gpu_training @pytest.mark.env_onecard def test_itemset_with_number(): diff --git a/tests/st/train/test_amp_overflow.py b/tests/st/train/test_amp_overflow.py index 4da3baf8675..073fe6f696b 100644 --- a/tests/st/train/test_amp_overflow.py +++ b/tests/st/train/test_amp_overflow.py @@ -36,6 +36,7 @@ class Net(nn.Cell): @pytest.mark.level0 @pytest.mark.platform_arm_ascend_training +@pytest.mark.platform_arm_ascend910b_training @pytest.mark.platform_x86_ascend_training @pytest.mark.platform_x86_gpu_training @pytest.mark.platform_x86_cpu_training diff --git a/tests/st/train/test_loss_scale_overflow.py b/tests/st/train/test_loss_scale_overflow.py index 085a8251f7c..fa353ea7953 100644 --- a/tests/st/train/test_loss_scale_overflow.py +++ b/tests/st/train/test_loss_scale_overflow.py @@ -37,6 +37,7 @@ class Net(nn.Cell): @pytest.mark.level0 @pytest.mark.platform_arm_ascend_training @pytest.mark.platform_x86_ascend_training +@pytest.mark.platform_arm_ascend910b_training @pytest.mark.platform_x86_gpu_training @pytest.mark.env_onecard @pytest.mark.parametrize('mode', [ms.GRAPH_MODE, ms.PYNATIVE_MODE])