[Cutlass] Fix int check in example tensor creation (#152306)

mlazos · pytorchmergebot · commit 72273bef9e96 · 2025-04-29T23:05:47.000Z
Pull Request resolved: #152306 Approved by: https://github.com/Skylion007, https://github.com/eellison ghstack dependencies: #152305
diff --git a/torch/_inductor/codegen/cuda/cutlass_lib_extensions/evt_extensions.py b/torch/_inductor/codegen/cuda/cutlass_lib_extensions/evt_extensions.py
@@ -69,10 +69,11 @@ def create_example_tensors(
         def cutlass_tensor_from_buffer(buffer: Buffer) -> CutlassTensor:
             shape = buffer.get_layout().size
             stride = buffer.get_layout().stride
-            assert all(isinstance(x, int) for x in buffer.get_layout().stride), (
+
+            assert all(x.is_integer for x in shape), (
                 f"{buffer.get_name()}'s shape {shape} contains symints which aren't supported for cutlass EVT"
             )
-            assert all(isinstance(x, int) for x in buffer.get_layout().stride), (
+            assert all(x.is_integer for x in stride), (
                 f"{buffer.get_name()}'s stride {stride} contains symints which aren't supported for cutlass EVT"
             )
             shape = tuple(int(x) for x in shape)

Original file line number	Diff line number	Diff line change
`@@ -69,10 +69,11 @@ def create_example_tensors(`
`69`	`69`	`def cutlass_tensor_from_buffer(buffer: Buffer) -> CutlassTensor:`
`70`	`70`	`shape = buffer.get_layout().size`
`71`	`71`	`stride = buffer.get_layout().stride`
`72`		`- assert all(isinstance(x, int) for x in buffer.get_layout().stride), (`
	`72`	`+`
	`73`	`+ assert all(x.is_integer for x in shape), (`
`73`	`74`	`f"{buffer.get_name()}'s shape {shape} contains symints which aren't supported for cutlass EVT"`
`74`	`75`	`)`
`75`		`- assert all(isinstance(x, int) for x in buffer.get_layout().stride), (`
	`76`	`+ assert all(x.is_integer for x in stride), (`
`76`	`77`	`f"{buffer.get_name()}'s stride {stride} contains symints which aren't supported for cutlass EVT"`
`77`	`78`	`)`
`78`	`79`	`shape = tuple(int(x) for x in shape)`