Merge 688e421184 into 8402a2acf4

2025-08-18 00:16:39 +08:00 · 2025-06-13 11:34:01 -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/device/"
+#define INCLUDE_PREFIX "mlx/backend/cuda/kernels/"

 constexpr const char* g_include_names[] = {
    INCLUDE_PREFIX "atomic_ops.cuh",