xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

offline st testcases for cuda-11.1

This commit is contained in:
xsmq 2021-05-25 14:43:32 +08:00
parent e1803a0938
commit a2e3eb1209
4 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/st/nccl/test_nccl_all.py
View File

@ -16,7 +16,7 @@ import os
import pytest import pytest
@pytest.mark.level0 @pytest.mark.level2
@pytest.mark.platform_x86_gpu_training @pytest.mark.platform_x86_gpu_training
@pytest.mark.env_single @pytest.mark.env_single
def test_nccl_lenet(): def test_nccl_lenet():

2
tests/st/networks/test_gpu_resnet.py
View File

@ -352,7 +352,7 @@ def test_trainTensor(num_classes=10, epoch=8, batch_size=1):
assert (losses[-1].asnumpy() < 1) assert (losses[-1].asnumpy() < 1)
@pytest.mark.level0 @pytest.mark.level2
@pytest.mark.platform_x86_gpu_training @pytest.mark.platform_x86_gpu_training
@pytest.mark.env_onecard @pytest.mark.env_onecard
def test_trainTensor_big_batchSize(num_classes=10, epoch=8, batch_size=338): def test_trainTensor_big_batchSize(num_classes=10, epoch=8, batch_size=338):

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

@ -105,7 +105,7 @@ def test_tensor_dot_fp32():
np.allclose(ms_result_np, np_result) np.allclose(ms_result_np, np_result)
@pytest.mark.level0 @pytest.mark.level2
@pytest.mark.platform_x86_gpu_training @pytest.mark.platform_x86_gpu_training
@pytest.mark.env_onecard @pytest.mark.env_onecard
def test_tensor_dot_fp16(): def test_tensor_dot_fp16():

2
tests/st/quantization/mobilenetv2_quant/test_mobilenetv2_quant_gpu.py
View File

@ -54,7 +54,7 @@ config_ascend_quant = ed({
dataset_path = "/home/workspace/mindspore_dataset/cifar-10-batches-bin/" dataset_path = "/home/workspace/mindspore_dataset/cifar-10-batches-bin/"
@pytest.mark.level0 @pytest.mark.level2
@pytest.mark.platform_x86_gpu_training @pytest.mark.platform_x86_gpu_training
@pytest.mark.env_single @pytest.mark.env_single
def test_mobilenetv2_quant(): def test_mobilenetv2_quant():