Skip to content

Commit 88642d4

Browse files
authored
Update int4_xpu_layout.py
1 parent 7d5ba43 commit 88642d4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchao/dtypes/uintx/int4_xpu_layout.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,7 @@ def _linear_fp_act_uint4_weight_int8_zero_impl(input_tensor, weight_tensor, bias
125125
orig_act_size = act_mat.size()
126126
orig_dtype = act_mat.dtype
127127

128-
act_mat = act_mat.reshape(-1, act_mat.shape[-1]).to(torch.bfloat16)
128+
act_mat = act_mat.reshape(-1, act_mat.shape[-1])
129129

130130
# groupwise int4 quantization
131131
groupsize = weight_tensor.block_size[1]

0 commit comments

Comments
 (0)