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 88from .annotate_quant_attrs import AnnotateQuantAttrs
99from .annotate_stack import AnnotateStack
1010from .annotate_unbind import AnnotateUnbind
11- from .convert_bmm_to_matmul import ConvertBmmToMatmul
1211from .convert_conv1d_to_conv2d import ConvertConv1dToConv2d
1312from .convert_square_to_pow import ConvertSquareToPow
1413from .decompose_any import DecomposeAny
4645 AnnotateQuantAttrs ,
4746 AnnotateStack ,
4847 AnnotateUnbind ,
49- ConvertBmmToMatmul ,
5048 ConvertConv1dToConv2d ,
5149 ConvertSquareToPow ,
5250 DecomposeAny ,
Load Diff This file was deleted.
Original file line number Diff line number Diff line change 1313 AnnotateQuantAttrs ,
1414 AnnotateStack ,
1515 AnnotateUnbind ,
16- ConvertBmmToMatmul ,
1716 ConvertConv1dToConv2d ,
1817 ConvertSquareToPow ,
1918 DecomposeAny ,
@@ -81,7 +80,6 @@ def get_capture_program_passes():
8180 (AnnotateQuantAttrs , True ),
8281 (AnnotateStack , True ),
8382 (AnnotateUnbind , True ),
84- (ConvertBmmToMatmul , True ),
8583 (ConvertConv1dToConv2d , True ),
8684 (DecomposeAny , True ),
8785 (DecomposeColIm , True ),
Original file line number Diff line number Diff line change @@ -64,7 +64,6 @@ def get_passes_dependency_for_capture_program():
6464 AnnotateQuantAttrs ,
6565 AnnotateStack ,
6666 AnnotateUnbind ,
67- ConvertBmmToMatmul ,
6867 ConvertConv1dToConv2d ,
6968 DecomposeAny ,
7069 DecomposeColIm ,
@@ -85,12 +84,10 @@ def get_passes_dependency_for_capture_program():
8584 AnnotateAdaptiveAvgPool1D : [RemoveRedundancy ],
8685 AnnotateQuantAttrs : [
8786 RecomposePixelUnshuffle ,
88- ConvertBmmToMatmul ,
8987 RemoveRedundancy ,
9088 ],
9189 AnnotateStack : [RemoveRedundancy ],
9290 AnnotateUnbind : [RemoveRedundancy ],
93- ConvertBmmToMatmul : [RecomposePixelUnshuffle ],
9491 DecomposeAny : [RemoveRedundancy ],
9592 DecomposeColIm : [FoldQDQ ],
9693 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]:
5050 torch .ops .aten .instance_norm .default ,
5151 torch .ops .aten .leaky_relu .default ,
5252 torch .ops .aten .linear .default ,
53+ torch .ops .aten .matmul .default ,
5354 torch .ops .aten .pixel_shuffle .default ,
5455 torch .ops .aten .pixel_unshuffle .default ,
5556 torch .ops .aten .prelu .default ,
You can’t perform that action at this time.
0 commit comments