File tree Expand file tree Collapse file tree 5 files changed +1
-83
lines changed Expand file tree Collapse file tree 5 files changed +1
-83
lines changed Original file line number Diff line number Diff line change 8
8
from .annotate_quant_attrs import AnnotateQuantAttrs
9
9
from .annotate_stack import AnnotateStack
10
10
from .annotate_unbind import AnnotateUnbind
11
- from .convert_bmm_to_matmul import ConvertBmmToMatmul
12
11
from .convert_conv1d_to_conv2d import ConvertConv1dToConv2d
13
12
from .convert_square_to_pow import ConvertSquareToPow
14
13
from .decompose_any import DecomposeAny
46
45
AnnotateQuantAttrs ,
47
46
AnnotateStack ,
48
47
AnnotateUnbind ,
49
- ConvertBmmToMatmul ,
50
48
ConvertConv1dToConv2d ,
51
49
ConvertSquareToPow ,
52
50
DecomposeAny ,
Load Diff This file was deleted.
Original file line number Diff line number Diff line change 13
13
AnnotateQuantAttrs ,
14
14
AnnotateStack ,
15
15
AnnotateUnbind ,
16
- ConvertBmmToMatmul ,
17
16
ConvertConv1dToConv2d ,
18
17
ConvertSquareToPow ,
19
18
DecomposeAny ,
@@ -81,7 +80,6 @@ def get_capture_program_passes():
81
80
(AnnotateQuantAttrs , True ),
82
81
(AnnotateStack , True ),
83
82
(AnnotateUnbind , True ),
84
- (ConvertBmmToMatmul , True ),
85
83
(ConvertConv1dToConv2d , True ),
86
84
(DecomposeAny , True ),
87
85
(DecomposeColIm , True ),
Original file line number Diff line number Diff line change @@ -64,7 +64,6 @@ def get_passes_dependency_for_capture_program():
64
64
AnnotateQuantAttrs ,
65
65
AnnotateStack ,
66
66
AnnotateUnbind ,
67
- ConvertBmmToMatmul ,
68
67
ConvertConv1dToConv2d ,
69
68
DecomposeAny ,
70
69
DecomposeColIm ,
@@ -85,12 +84,10 @@ def get_passes_dependency_for_capture_program():
85
84
AnnotateAdaptiveAvgPool1D : [RemoveRedundancy ],
86
85
AnnotateQuantAttrs : [
87
86
RecomposePixelUnshuffle ,
88
- ConvertBmmToMatmul ,
89
87
RemoveRedundancy ,
90
88
],
91
89
AnnotateStack : [RemoveRedundancy ],
92
90
AnnotateUnbind : [RemoveRedundancy ],
93
- ConvertBmmToMatmul : [RecomposePixelUnshuffle ],
94
91
DecomposeAny : [RemoveRedundancy ],
95
92
DecomposeColIm : [FoldQDQ ],
96
93
DecomposeLinalgVectorNorm : [RemoveRedundancy ],
Original file line number Diff line number Diff line change @@ -50,6 +50,7 @@ def get_skip_decomp_table() -> List[torch._ops.OperatorBase]:
50
50
torch .ops .aten .instance_norm .default ,
51
51
torch .ops .aten .leaky_relu .default ,
52
52
torch .ops .aten .linear .default ,
53
+ torch .ops .aten .matmul .default ,
53
54
torch .ops .aten .pixel_shuffle .default ,
54
55
torch .ops .aten .pixel_unshuffle .default ,
55
56
torch .ops .aten .prelu .default ,
You can’t perform that action at this time.
0 commit comments