Skip to content

Commit 680e59d

Browse files
committed
fixing experimental test
Summary: Test Plan: Reviewers: Subscribers: Tasks: Tags:
1 parent 900fbba commit 680e59d

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

torchao/dtypes/uintx/packed_linear_int8_dynamic_activation_intx_weight_layout.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -359,6 +359,9 @@ def _impl_2d_aten(input_tensor, weight_tensor):
359359

360360
m, k = input_tensor.shape
361361
n, k_ = weight_tensor.shape
362+
if m==0: # handling for empty input
363+
return input_tensor
364+
362365
assert k_ == k
363366
group_size = weight_tensor.tensor_impl.get_layout().group_size
364367
packed_weight = weight_tensor.tensor_impl.packed_weight

0 commit comments

Comments
 (0)