Commit 9d18531
authored
Arm backend: Fix bmm quantization bug (pytorch#19798)
bmm nodes are now forwarded to ArmPass in stead of ExportPass.
This fixes an issue where _call_quantized_bmm_without_fake_kernel()
does not get called, leading to dtype mismatch error
cc @digantdesai @freddan80 @per @zingo @oscarandersson8218 @mansnils
@Sebastian-Larsson @robell @rascani
Signed-off-by: Christoffer J.L <christoffer.johanssonlundqvist@arm.com>1 parent b1446cc commit 9d18531
1 file changed
Lines changed: 1 addition & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
126 | 126 | | |
127 | 127 | | |
128 | 128 | | |
129 | | - | |
| 129 | + | |
0 commit comments