[Cutlass] Fix int check in example tensor creation

mlazos · mlazos · commit 966689bf29fe · 2025-04-28T17:13:06.000-07:00
ghstack-source-id: 907aa83 Pull Request resolved: #152306
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)`