fixing experimental CI

Summary:

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:

Author: HDCharles

torchao/dtypes/uintx/packed_linear_int8_dynamic_activation_intx_weight_layout.py

@@ -359,8 +359,6 @@ def _impl_2d_aten(input_tensor, weight_tensor):
 
         m, k = input_tensor.shape
         n, k_ = weight_tensor.shape
-        if m==0: # handling for empty input
-            return input_tensor
 
         assert k_ == k
         group_size = weight_tensor.tensor_impl.get_layout().group_size
@@ -369,6 +367,9 @@ def _impl_2d_aten(input_tensor, weight_tensor):
             input_tensor, packed_weight, group_size, k, n
         )
 
+    if input_tensor.numel() == 0:
+        return input_tensor
+
     target = weight_tensor.tensor_impl.get_layout().target
 
     if weight_tensor.tensor_impl.get_layout().has_bias: