multi arch kernel

JacobSzwejbka · JacobSzwejbka · commit 18b95354875c · 2025-10-20T09:54:40.000-07:00
diff --git a/backends/cuda/cuda_backend.py b/backends/cuda/cuda_backend.py
@@ -180,6 +180,7 @@ def preprocess(
                     "aot_inductor.aoti_shim_library": "aoti_cuda",
                     "aot_inductor.aoti_shim_library_path": shim_library_path,
                     "aot_inductor.precompile_headers": False,
+                    "aot_inductor.emit_multi_arch_kernel": True,
                 }
             )
 

Original file line number	Diff line number	Diff line change
`@@ -180,6 +180,7 @@ def preprocess(`
`180`	`180`	`"aot_inductor.aoti_shim_library": "aoti_cuda",`
`181`	`181`	`"aot_inductor.aoti_shim_library_path": shim_library_path,`
`182`	`182`	`"aot_inductor.precompile_headers": False,`
	`183`	`+ "aot_inductor.emit_multi_arch_kernel": True,`
`183`	`184`	`}`
`184`	`185`	`)`
`185`	`186`