From f2d0ea06072c29b1aa28c3fcbe10501ce2be44bc Mon Sep 17 00:00:00 2001 From: Awni Hannun Date: Fri, 13 Jun 2025 15:01:16 -0700 Subject: [PATCH] fix cuda jit --- 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",