From 82fff8535595b9da6ada59f0f912becee861fbb4 Mon Sep 17 00:00:00 2001 From: Awni Hannun Date: Sun, 27 Jul 2025 06:31:30 -0700 Subject: [PATCH] comment --- mlx/backend/cuda/device.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mlx/backend/cuda/device.cpp b/mlx/backend/cuda/device.cpp index ba58b9aec..38bc73f7f 100644 --- a/mlx/backend/cuda/device.cpp +++ b/mlx/backend/cuda/device.cpp @@ -56,7 +56,8 @@ Device::Device(int device) : device_(device) { // The cudnn handle is used by Convolution. CHECK_CUDNN_ERROR(cudnnCreate(&cudnn_)); - // Ensure the jit module cache is initialized + // Initialize the jit module cache here ensures it is not + // unloaded before any evaluation is done get_jit_module_cache(); }