$ 0 0 Could you try set these variables and run test againexport MLX_QP_ALLOC_TYPE=HUGEexport MLX_CP_ALLOC_TYPE=HUGE