Skip to content

Commit

Permalink
avoid .to() followed by inplace mutation to appease export
Browse files Browse the repository at this point in the history
ghstack-source-id: b979b7637c8d062b5f5a61c56e1e78edf1b70290
Pull Request resolved: #1387
  • Loading branch information
bdhirsh committed Dec 6, 2024
1 parent 8a805d0 commit e57993d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torchao/dtypes/uintx/plain_layout.py
Original file line number Diff line number Diff line change
Expand Up @@ -264,5 +264,5 @@ def _linear_int8_act_int8_weight_impl(input_tensor, weight_tensor, bias):
output_dtype = input_tensor.dtype
y = y.to(output_dtype)
if bias is not None:
y += bias
y = y + bias
return y

0 comments on commit e57993d

Please sign in to comment.