github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-04 13:19:43 +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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 0 additions and 3 deletions
Show Stats 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
@run_on_environment_flag(name='AUTO_PARALLEL')
def test_getitem_function_handler():
model = GetItemModel()
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
@run_on_environment_flag(name='AUTO_PARALLEL')
def test_reshape_handler():
model = ReshapeModel()
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
@run_on_environment_flag(name='AUTO_PARALLEL')
def test_elementwise_handler():
model = ReLuModel()
tracer = ColoTracer()