From 6871e2eeb7f058d4a4dd1c76cb0464fd51aa11ba Mon Sep 17 00:00:00 2001 From: Awni Hannun Date: Fri, 13 Jun 2025 19:21:46 -0700 Subject: [PATCH] fix cuda jit (#2287) --- mlx/backend/cuda/jit_module.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mlx/backend/cuda/jit_module.cpp b/mlx/backend/cuda/jit_module.cpp index b8be103cc..8a033523c 100644 --- a/mlx/backend/cuda/jit_module.cpp +++ b/mlx/backend/cuda/jit_module.cpp @@ -145,7 +145,7 @@ bool compiler_supports_device_sass(Device& device) { } } -#define INCLUDE_PREFIX "mlx/backend/cuda/kernels/" +#define INCLUDE_PREFIX "mlx/backend/cuda/device/" constexpr const char* g_include_names[] = { INCLUDE_PREFIX "atomic_ops.cuh",