mirror of https://github.com/hpcaitech/ColossalAI
[autoparallel] recovered skipped test cases (#1748)
parent
474111ecb5
commit
b80b6eaa88
|
@ -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()
|
||||
|
|
|
@ -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()
|
||||
|
|
|
@ -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()
|
||||
|
|
Loading…
Reference in New Issue