@@ -27,7 +27,7 @@ class TrainPaddleConfig: | |||||
@pytest.mark.parametrize("driver,device", [("paddle", "cpu"), ("paddle", 1), ("fleet", [0, 1])]) | @pytest.mark.parametrize("driver,device", [("paddle", "cpu"), ("paddle", 1), ("fleet", [0, 1])]) | ||||
# @pytest.mark.parametrize("driver,device", [("fleet", [0, 1])]) | # @pytest.mark.parametrize("driver,device", [("fleet", [0, 1])]) | ||||
@pytest.mark.parametrize("callbacks", [[RichCallback(5)]]) | @pytest.mark.parametrize("callbacks", [[RichCallback(5)]]) | ||||
@pytest.mark.paddle | |||||
@pytest.mark.paddledist | |||||
@magic_argv_env_context | @magic_argv_env_context | ||||
def test_trainer_paddle( | def test_trainer_paddle( | ||||
driver, | driver, | ||||
@@ -19,7 +19,7 @@ if _NEED_IMPORT_PADDLE: | |||||
import paddle | import paddle | ||||
import paddle.distributed as dist | import paddle.distributed as dist | ||||
@pytest.mark.paddle | |||||
@pytest.mark.paddledist | |||||
class TestDistUtilsTools: | class TestDistUtilsTools: | ||||
""" | """ | ||||
测试一些工具函数 | 测试一些工具函数 | ||||
@@ -79,14 +79,13 @@ class TestDistUtilsTools: | |||||
assert res["int"] == paddle_dict["int"] | assert res["int"] == paddle_dict["int"] | ||||
assert res["string"] == paddle_dict["string"] | assert res["string"] == paddle_dict["string"] | ||||
@pytest.mark.paddle | |||||
@pytest.mark.paddledist | |||||
class TestAllGatherAndBroadCast: | class TestAllGatherAndBroadCast: | ||||
@classmethod | @classmethod | ||||
def setup_class(cls): | def setup_class(cls): | ||||
devices = [0,1,2] | devices = [0,1,2] | ||||
output_from_new_proc = "only_error" | |||||
output_from_new_proc = "all" | |||||
launcher = FleetLauncher(devices=devices, output_from_new_proc=output_from_new_proc) | launcher = FleetLauncher(devices=devices, output_from_new_proc=output_from_new_proc) | ||||
cls.local_rank = int(os.getenv("PADDLE_RANK_IN_NODE", "0")) | cls.local_rank = int(os.getenv("PADDLE_RANK_IN_NODE", "0")) | ||||
@@ -39,7 +39,7 @@ def generate_driver(num_labels, feature_dimension, device=[0,1], fp16=False, out | |||||
# | # | ||||
############################################################################ | ############################################################################ | ||||
@pytest.mark.paddle | |||||
@pytest.mark.paddledist | |||||
class TestFleetDriverFunction: | class TestFleetDriverFunction: | ||||
""" | """ | ||||
测试 PaddleFleetDriver 一些简单函数的测试类,基本都是测试能否运行、是否存在 import 错误等问题 | 测试 PaddleFleetDriver 一些简单函数的测试类,基本都是测试能否运行、是否存在 import 错误等问题 | ||||
@@ -147,7 +147,7 @@ class TestFleetDriverFunction: | |||||
# | # | ||||
############################################################################ | ############################################################################ | ||||
@pytest.mark.paddle | |||||
@pytest.mark.paddledist | |||||
class TestSetDistReproDataloader: | class TestSetDistReproDataloader: | ||||
@classmethod | @classmethod | ||||
@@ -521,7 +521,7 @@ class TestSetDistReproDataloader: | |||||
# | # | ||||
############################################################################ | ############################################################################ | ||||
@pytest.mark.paddle | |||||
@pytest.mark.paddledist | |||||
class TestSaveLoad: | class TestSaveLoad: | ||||
""" | """ | ||||
测试多卡情况下 save 和 load 相关函数的表现 | 测试多卡情况下 save 和 load 相关函数的表现 | ||||
@@ -2,5 +2,6 @@ | |||||
markers = | markers = | ||||
torch | torch | ||||
paddle | paddle | ||||
paddledist | |||||
jittor | jittor | ||||
torchpaddle | torchpaddle |