Temporarily set the default MLX device
Arguments
- device
"gpu","cpu", or anmlx_streamcreated viamlx_new_stream().- code
Expression to evaluate while
deviceis active.
Examples
with_default_device("cpu", x <- mlx_vector(1:10))
Temporarily set the default MLX device
"gpu", "cpu", or an mlx_stream created via mlx_new_stream().
Expression to evaluate while device is active.
with_default_device("cpu", x <- mlx_vector(1:10))