Merge 688e421184 into 6871e2eeb7

fix cuda jit (#2287 )
2025-08-07 12:30:50 +08:00 · 2025-06-13 21:55:29 -07:00 · 2025-06-13 19:21:46 -07:00
1 changed files with 1 additions and 1 deletions
--- 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",
Author	SHA1	Message	Date
Awni Hannun	c5be966863	Merge `688e421184` into `6871e2eeb7`	2025-06-13 21:55:29 -07:00
Awni Hannun	6871e2eeb7	fix cuda jit (#2287 )	2025-06-13 19:21:46 -07:00