Skip to content

Commit ba635b1

Browse files
committed
minor formatting
Signed-off-by: Andrea Fasoli <andrea.fasoli@ibm.com>
1 parent 2fe51f1 commit ba635b1

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

fms_mo/aiu_addons/fp8/fp8_spyre_op.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,8 +50,7 @@ def _scaled_mm_cpu_out(
5050
mat2 = (mat2.to(dtype=out_dtype) * scale2).to(dtype=out_dtype)
5151

5252
if bias is not None:
53-
bias_converted = bias.to(dtype=out_dtype)
54-
ret = torch.addmm(bias_converted, mat1, mat2)
53+
ret = torch.addmm(bias.to(dtype=out_dtype), mat1, mat2)
5554
else:
5655
ret = torch.mm(mat1, mat2)
5756

0 commit comments

Comments
 (0)