Skip to content

Allow configuring the CPU JIT cache directory - #4714

Open
cerisier wants to merge 2 commits into
ml-explore:mainfrom
cerisier:mlx/cpu-jit-cache
Open

cerisier wants to merge 2 commits into
ml-explore:mainfrom
cerisier:mlx/cpu-jit-cache

Conversation

@cerisier

@cerisier cerisier commented Oct 16, 2026 •

Copy link
Copy Markdown

Add MLX_CPU_CACHE_DIR so hermetic build and test setups such as Bazel can explicitly control where CPU JIT artifacts are stored.

I reused a similar approach for MLX_PTX_CACHE_DIR for CUDA so hopefully this makes ense.

@cerisier
cerisier marked this pull request as ready for review October 16, 2026 19:24
@zcbenz
zcbenz force-pushed the mlx/cpu-jit-cache branch from b5e7ef5 to 4063c21 Compare October 19, 2026 00:56
@zcbenz
zcbenz force-pushed the mlx/cpu-jit-cache branch from 4063c21 to 9b5df75 Compare October 19, 2026 03:42

This branch has not been deployed

No deployments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants