[BUG] crash on qr followed by diag

Describe the bug

mlx crashes on my M2 Macbook Air with the following code:

import mlx.core as mx

x = mx.random.normal((10000, 10000))
q, r = mx.linalg.qr(x, stream=mx.cpu)
mx.eval(mx.diag(r))

giving

libc++abi: terminating due to uncaught exception of type std::runtime_error: [METAL] Command buffer execution failed: Caused GPU Timeout Error (00000002:kIOGPUCommandBufferCallbackErrorTimeout)

The following variant does not crash:

import mlx.core as mx

x = mx.random.normal((10000, 10000))
q, r = mx.linalg.qr(x, stream=mx.cpu)
mx.eval(r)
mx.eval(mx.diag(r))

Expected behavior
I'd expect the operation not to fail, given that the intermediate evaluation lets it succeed. But if this is a simple oom error then sorry for the false report.

Desktop:

• OS Version: Tahoe 26.4.1
• Mlx Version: 0.31.1

Reproduce the reported Python snippet on an M2 Mac with MLX 0.31.1, comparing the failing qr-to-diag sequence with the variant that evaluates r first. Trace the QR, diag, and evaluation behavior to identify why the command buffer times out; done means the original sequence completes without a crash.