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 Packages Projects Releases Wiki Activity

[CUDA] Fix segfault on exit (#2424)

Browse Source 
* fix cuda segfault on exit

* comment
This commit is contained in:
Awni Hannun
2025-07-27 08:08:13 -07:00
committed by GitHub
parent 4ad53414dd
commit b9e88fb976
3 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mlx/backend/cuda/jit_module.h
View File

@@ -99,6 +99,8 @@ class JitModule {
std::unordered_map<std::string, CUfunction> kernels_;
};
std::unordered_map<std::string, JitModule>& get_jit_module_cache();
JitModule& get_jit_module(
const mlx::core::Device& device,
const std::string& name,