Update on "[Cutlass] Fix tests"

mlazos · mlazos · commit 96924f30dcef · 2025-05-08T15:08:25.000-07:00
cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx ipiszy chenyang78 kadeng muchulee8 amjames chauhang aakhundov

[ghstack-poisoned]
diff --git a/test/inductor/test_cutlass_evt.py b/test/inductor/test_cutlass_evt.py
@@ -208,7 +208,8 @@ def inner_fn_buf4(index):
 
             self.assertExpectedInline(
                 str(result),
-                """Unsupported indexing for buf0 with index 200*i0 + 60000*i1 + i2 and strides [200, 60000, 1]""",
+                """Unsupported indexing for buf0 with index 200*i0 + 60000*i1 + i2, \
+index strides [200, 60000, 1], and layout stride [60000, 200, 1]""",
             )
 
     @unittest.skipIf(not SM90OrLater, "need sm_90")
@@ -359,10 +360,7 @@ def test_example_tensor_creation(self):
     @unittest.skipIf(not SM90OrLater, "need sm_90")
     @unittest.skipIf(not try_import_cutlass(), "requires cutlass")
     def test_evt_argument_codegen(self):
-        from torch._inductor.codegen.cuda.cuda_env import get_cuda_arch
-
-        cuda_arch = int(get_cuda_arch())  # type: ignore[arg-type]
-        epilogue_functor = _trace(BIAS_CODE, EXAMPLE_TENSORS, cuda_arch)
+        epilogue_functor = _trace(BIAS_CODE, EXAMPLE_TENSORS)
 
         self.assertExpectedInline(
             _render_argument_type(
