Skip to content

Commit e434056

Browse files
authored
use all fuse_and_eliminate passes as default passes for optimize (#124)
Signed-off-by: haoshengqiang <haoshengqiang79@163.com>
1 parent 8a5295d commit e434056

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

onnxoptimizer/__init__.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -34,11 +34,7 @@ def optimize(model, passes=None, fixed_point=False): # type: (ModelProto, Optio
3434
"""
3535

3636
if passes is None:
37-
print('WARNING: defualt optimization passes will be enlarged to all fuse and elimination passes in the next version')
38-
passes = ['eliminate_nop_transpose',
39-
'eliminate_nop_pad',
40-
'fuse_consecutive_transposes',
41-
'fuse_transpose_into_gemm']
37+
passes = get_fuse_and_elimination_passes()
4238
if not isinstance(model, ModelProto):
4339
raise ValueError(
4440
'Optimizer only accepts ModelProto, incorrect type: {}'.format(type(model)))

0 commit comments

Comments
 (0)