@@ -337,8 +337,8 @@ def get_decomp_table() -> Dict[torch._ops.OperatorBase, Callable]:
337
337
torch .ops .pt2e_quant .quantize_affine .default ,
338
338
torch .ops .pt2e_quant .dequantize_affine .default ,
339
339
torch .ops .aten ._safe_softmax .default ,
340
- torch .ops .aten .stack .default , # TODO: Might need to remove this later due to Mimi. QNN does not support int io for stack op.
341
- torch .ops .aten .unbind .int ,
340
+ # torch.ops.aten.stack.default, # TODO: Might need to remove this later due to Mimi. QNN does not support int io for stack op.
341
+ # torch.ops.aten.unbind.int,
342
342
]
343
343
344
344
remove_decompositions (source_decompositions , skip_decompositions )
@@ -358,7 +358,7 @@ def get_capture_program_passes():
358
358
# The second value in each tuple in `default_passes_and_setting` indicates whether the corresponding pass is activated by default.
359
359
# If a pass is activated, it will be executed by default.
360
360
default_passes_and_setting = [
361
- (AnnotateDecomposed , False ),
361
+ (AnnotateDecomposed , True ),
362
362
(AnnotateQuantAttrs , True ),
363
363
(ConstantI64toI32 , True ),
364
364
(ConvertBmmToMatmul , True ),
0 commit comments