From b80b6eaa8870ab24d677b9a703c7f4a663d87ee2 Mon Sep 17 00:00:00 2001 From: Frank Lee Date: Thu, 20 Oct 2022 16:37:33 +0800 Subject: [PATCH] [autoparallel] recovered skipped test cases (#1748) --- .../test_tensor_shard/test_node_handler/test_getitem_handler.py | 1 - .../test_tensor_shard/test_node_handler/test_reshape_handler.py | 1 - .../test_node_handler/test_unary_element_wise_handler.py | 1 - 3 files changed, 3 deletions(-) diff --git a/tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_getitem_handler.py b/tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_getitem_handler.py index d185eb6db..5f7c469bc 100644 --- a/tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_getitem_handler.py +++ b/tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_getitem_handler.py @@ -21,7 +21,6 @@ class GetItemModel(nn.Module): return x -@run_on_environment_flag(name='AUTO_PARALLEL') def test_getitem_function_handler(): model = GetItemModel() tracer = ColoTracer() diff --git a/tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_reshape_handler.py b/tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_reshape_handler.py index de277002b..613f8f3d0 100644 --- a/tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_reshape_handler.py +++ b/tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_reshape_handler.py @@ -20,7 +20,6 @@ class ReshapeModel(nn.Module): return reshape_node -@run_on_environment_flag(name='AUTO_PARALLEL') def test_reshape_handler(): model = ReshapeModel() tracer = ColoTracer() diff --git a/tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_unary_element_wise_handler.py b/tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_unary_element_wise_handler.py index a861cb7f5..e4d12cd12 100644 --- a/tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_unary_element_wise_handler.py +++ b/tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_unary_element_wise_handler.py @@ -22,7 +22,6 @@ class ReLuModel(nn.Module): return relu_node -@run_on_environment_flag(name='AUTO_PARALLEL') def test_elementwise_handler(): model = ReLuModel() tracer = ColoTracer()