zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

comment

Browse Source
This commit is contained in:
Awni Hannun
2025-07-27 06:31:30 -07:00
parent 254c0f56c3
commit 82fff85355
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
mlx/backend/cuda/device.cpp
View File

@@ -56,7 +56,8 @@ Device::Device(int device) : device_(device) {
// The cudnn handle is used by Convolution. // The cudnn handle is used by Convolution.
CHECK_CUDNN_ERROR(cudnnCreate(&cudnn_)); 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(); get_jit_module_cache();
} }