Daniel Hiltgen 570679c9e0 mlx: update and fix CUDA JIT packaging (#16871) ...

Bump MLX to the latest selected upstream ref and update the MLX/imagegen
wrappers and tests for the new API behavior.

Fix the CUDA MLX archive so runtime NVRTC kernels work after deployment:
package CUTE/CUTLASS headers, include the CUDA runtime header closure, and
stage a coherent CUDA-toolkit-matched CCCL tree instead of MLX's fetched CCCL
for CUDA payloads. The previous archive could build successfully but crash at
runtime due to missing or incompatible JIT headers.

2026-06-24 10:36:02 -07:00

3.8 KiB

Executable File

Raw Permalink Blame History

View Raw