ThreadLocalStream in C++ (#3405)

2026-04-15 15:46:11 -07:00
parent fd8e849e26
commit dec6b4d10f
9 changed files with 80 additions and 43 deletions
@@ -25,7 +25,7 @@ class TestThreads(mlx_tests.MLXTestCase):
        t1.join()
        t2.join()

-        test_stream = mx.ThreadLocalStream(mx.default_device())
+        test_stream = mx.new_thread_local_stream(mx.default_device())

        def test_success():
            with mx.stream(test_stream):