-
Notifications
You must be signed in to change notification settings - Fork 377
Open
Labels
pt2e_quantpt2 export quantization (prepare_pt2e, convert_pt2e)pt2 export quantization (prepare_pt2e, convert_pt2e)
Description
🚀 The feature, motivation and pitch
There is BatchNorm + Convolution fusion happening in prepare_pt2e function call. I don't think it's good practice to introduce optimizations randomly into the pipeline (prepare_pt2e function call) because this might be a side effect for the user of an API.
I'm proposing removal of this optimization and letting user peform this optimization on his own, before quantization.
Follow-up from: pytorch/executorch#10579 (comment)
cc @jerryzh168 @jianyuh @raghuramank100 @jamesr66a @vkuzo @jgong5 @Xia-Weiwen @leslie-fang-intel @msaroufim @digantdesai
Alternatives
No response
Additional context
No response
Metadata
Metadata
Assignees
Labels
pt2e_quantpt2 export quantization (prepare_pt2e, convert_pt2e)pt2 export quantization (prepare_pt2e, convert_pt2e)