diff --git a/‎tilelang/jit/__init__.py‎
Lines changed: 1 addition & 1 deletion b/‎tilelang/jit/__init__.py‎
Lines changed: 1 addition & 1 deletion
@@ -410,7 +410,7 @@ def compile(self, *args: _P.args, **kwargs: _P.kwargs) -> _Ret:
                 func_name = getattr(self.func, "__name__", "jit_kernel")
 
             # cutedsl emits python executor not `c`
-            is_cutedsl = self.execution_backend == "cutedsl"
+            is_cutedsl = (self.execution_backend or self.target) == "cutedsl"
             kernel_suffix = "py" if is_cutedsl else "c"
             kernel_file = f"tilelang_jit_kernel_{func_name}.{kernel_suffix}"