From aa07429bada064b1913031057d230d3a9b5663b0 Mon Sep 17 00:00:00 2001 From: Awni Hannun Date: Thu, 12 Jun 2025 17:48:05 -0700 Subject: [PATCH] Fix cuda build (#2284) --- mlx/backend/cuda/CMakeLists.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mlx/backend/cuda/CMakeLists.txt b/mlx/backend/cuda/CMakeLists.txt index 3e7b859a6..9b12d84a9 100644 --- a/mlx/backend/cuda/CMakeLists.txt +++ b/mlx/backend/cuda/CMakeLists.txt @@ -44,8 +44,8 @@ target_compile_definitions(mlx PRIVATE MLX_USE_CUDA) file( GLOB MLX_JIT_SOURCES RELATIVE ${CMAKE_CURRENT_SOURCE_DIR} - "${CMAKE_CURRENT_SOURCE_DIR}/kernels/*.h" - "${CMAKE_CURRENT_SOURCE_DIR}/kernels/*.cuh") + "${CMAKE_CURRENT_SOURCE_DIR}/device/*.h" + "${CMAKE_CURRENT_SOURCE_DIR}/device/*.cuh") string(JOIN ":" MLX_JIT_SOURCES_ARG ${MLX_JIT_SOURCES}) add_custom_command( OUTPUT gen/cuda_jit_sources.h