Skip to content

Commit 9d18531

Browse files
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

File tree

backends/arm/_passes/replace_scalar_with_tensor_pass.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -126,4 +126,4 @@ def call_operator(self, op, args, kwargs, meta):
126126
return super().call_operator(op, args, kwargs, meta)
127127
else:
128128
# Do not handle; forward unchanged.
129-
return ExportPass.call_operator(self, op, args, kwargs, meta)
129+
return ArmPass.call_operator(self, op, args, kwargs, meta)

0 commit comments

Comments
 (0)