diff --git a/python/CuTeDSL/cutlass/base_dsl/jit_executor.py b/python/CuTeDSL/cutlass/base_dsl/jit_executor.py index 83268009c8..55377a4352 100644 --- a/python/CuTeDSL/cutlass/base_dsl/jit_executor.py +++ b/python/CuTeDSL/cutlass/base_dsl/jit_executor.py @@ -257,7 +257,7 @@ def run_compiled_program(self, exe_args): raise DSLRuntimeError(f"💥💥💥 Runtime Crash 💥💥💥", cause=e) def update_jit_cuda_modules(self, kernel_symbols): - # preload cuda module from compiled cubin in ir and store to jit_executor.kernels. + # preload cuda module from compiled cubin in ir and store to jit_executor.cuda_modules. if len(kernel_symbols) > 0: extra_args = [] module = self.ir_module