fix test to add in layout kwarg

Author: jcaip

test/sparsity/test_sparse_api.py

@@ -180,7 +180,9 @@ def test_fp8_cutlass_sparse_lowering_op_to(self):
             quantize_(model, Float8DynamicActivationFloat8SemiSparseWeightConfig())
 
             original = torch.ops.aten.to.dtype_layout(
-                model.weight.original_weight_tensor.tensor_impl, dtype=torch.float
+                model.weight.original_weight_tensor.tensor_impl,
+                dtype=torch.float,
+                layout=torch.strided,
             )
             torch.testing.assert_close(expected, original, atol=1e-1, rtol=1e-1)
 

torchao/dtypes/floatx/cutlass_semi_sparse_layout.py

@@ -106,7 +106,11 @@ def __torch_dispatch__(cls, func, types, args, kwargs):
             )
         elif func is aten.to.dtype_layout:
             dense, scale, _ = args[0].get_plain()
-            dense = dense.to(*args[1:], **kwargs)
+            dense = dense.to(
+                *args[1:],
+                dtype=kwargs.get("dtype", dense.dtype),
+                device=kwargs.get("device", dense.device),
+            )
             return scale * dense
 
         raise NotImplementedError(