github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-11 13:59:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

[autoparallel] recovered skipped test cases (#1748)

Browse Source
This commit is contained in:
Frank Lee
2022-10-20 16:37:33 +08:00
committed by GitHub
parent 474111ecb5
commit b80b6eaa88
3 changed files with 0 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_getitem_handler.py
View File

@@ -21,7 +21,6 @@ class GetItemModel(nn.Module):
return x return x
@run_on_environment_flag(name='AUTO_PARALLEL')
def test_getitem_function_handler(): def test_getitem_function_handler():
model = GetItemModel() model = GetItemModel()
tracer = ColoTracer() tracer = ColoTracer()

1
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_reshape_handler.py
View File

@@ -20,7 +20,6 @@ class ReshapeModel(nn.Module):
return reshape_node return reshape_node
@run_on_environment_flag(name='AUTO_PARALLEL')
def test_reshape_handler(): def test_reshape_handler():
model = ReshapeModel() model = ReshapeModel()
tracer = ColoTracer() tracer = ColoTracer()

1
tests/test_auto_parallel/test_tensor_shard/test_node_handler/test_unary_element_wise_handler.py
View File

@@ -22,7 +22,6 @@ class ReLuModel(nn.Module):
return relu_node return relu_node
@run_on_environment_flag(name='AUTO_PARALLEL')
def test_elementwise_handler(): def test_elementwise_handler():
model = ReLuModel() model = ReLuModel()
tracer = ColoTracer() tracer = ColoTracer()