From c422050ca734d3ba1b13547c02b01b9a68ae7663 Mon Sep 17 00:00:00 2001 From: Cheng Date: Sun, 17 Aug 2025 19:13:01 +0900 Subject: [PATCH] Update cuDNN Frontend to v1.14 (#2505) --- mlx/backend/cuda/CMakeLists.txt | 2 +- mlx/backend/cuda/conv.cpp | 3 --- 2 files changed, 1 insertion(+), 4 deletions(-) diff --git a/mlx/backend/cuda/CMakeLists.txt b/mlx/backend/cuda/CMakeLists.txt index 0d526400d..38708be5e 100644 --- a/mlx/backend/cuda/CMakeLists.txt +++ b/mlx/backend/cuda/CMakeLists.txt @@ -149,7 +149,7 @@ target_link_libraries(mlx PRIVATE CUDA::nvrtc CUDA::cuda_driver) FetchContent_Declare( cudnn GIT_REPOSITORY https://github.com/NVIDIA/cudnn-frontend.git - GIT_TAG v1.12.1 + GIT_TAG v1.14.0 GIT_SHALLOW TRUE EXCLUDE_FROM_ALL) set(CUDNN_FRONTEND_SKIP_JSON_LIB ON) diff --git a/mlx/backend/cuda/conv.cpp b/mlx/backend/cuda/conv.cpp index 3d7ef60bc..29d4803a8 100644 --- a/mlx/backend/cuda/conv.cpp +++ b/mlx/backend/cuda/conv.cpp @@ -7,9 +7,6 @@ #include "mlx/dtype_utils.h" #include "mlx/primitives.h" -// cudnn_frontend.h redefines this macro. -#undef CHECK_CUDA_ERROR - #include #include #include