Synchronize MLX execution

Waits for outstanding operations on the specified device or stream to complete.

Usage

mlx_synchronize(device = c("gpu", "cpu"))

Arguments

device

Device identifier ("gpu" or "cpu") or an mlx_stream created by mlx_new_stream().

See also

mlx.core.default_device

Examples

x <- as_mlx(matrix(1:4, 2, 2))
mlx_synchronize("gpu")
stream <- mlx_new_stream()
mlx_synchronize(stream)