Skip to content

Commit b5a7f81

Browse files
manuelcandalesfacebook-github-bot
authored andcommitted
Small fix to addmm broadcasting implementation
Reviewed By: kirklandsign Differential Revision: D48478732 fbshipit-source-id: 7fa7f351a4c6afa81e6bb64d3986573e396bb9f8
1 parent ce5d965 commit b5a7f81

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

kernels/portable/cpu/op_addmm.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ Tensor& addmm_out(
8484
[alpha_val, beta_val](const CTYPE val_a, const CTYPE val_b) {
8585
CTYPE a_casted = static_cast<CTYPE>(val_a);
8686
CTYPE b_casted = static_cast<CTYPE>(val_b);
87-
CTYPE value = a_casted + alpha_val * b_casted * beta_val;
87+
CTYPE value = a_casted * alpha_val + b_casted * beta_val;
8888

8989
return value;
9090
},

0 commit comments

Comments
 (0)