zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-24 17:31:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
8a0677d56d
mlx/python/src
History
Awni Hannun 8a0677d56d
Shared events for synchronization + async eval (#998) 
* more async eval

* fix rebase

* try correct async eval

* fix async

* more tests for async eval

* use shared events for synchronization

* comment + cleanup

* with autorelease pool

* fix no metal build

* fix compile

* fix patch

* don't eval if asyn evale'd

* don't use is_evaled

* comments

* more multi stream tests

* try and cleanup use of is_evaled

* use a status flag
2024-04-17 06:16:02 -07:00
..
array.cpp Shared events for synchronization + async eval (#998) 2024-04-17 06:16:02 -07:00
buffer.h Shared events for synchronization + async eval (#998) 2024-04-17 06:16:02 -07:00
CMakeLists.txt Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
constants.cpp Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
convert.cpp Shared events for synchronization + async eval (#998) 2024-04-17 06:16:02 -07:00
convert.h Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
device.cpp Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
fast.cpp post nanobind docs fixes and some updates (#889) 2024-03-24 15:03:27 -07:00
fft.cpp Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
indexing.cpp Indexing bug fix (#947) 2024-04-01 12:18:50 -07:00
indexing.h Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
linalg.cpp Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
load.cpp Do not store iostream in shared_ptr (#872) 2024-03-22 06:54:45 -07:00
load.h Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
metal.cpp Improve profiling with gpu tracing (#969) 2024-04-07 21:47:43 -07:00
mlx.cpp Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
ops.cpp Masked mm (#978) 2024-04-16 14:45:39 -07:00
random.cpp Implementation of mlx.random.multivariate_normal (#502) (#877) 2024-04-09 13:50:12 -07:00
stream.cpp Switch to nanobind (#839) 2024-03-18 20:12:25 -07:00
transforms.cpp Shared events for synchronization + async eval (#998) 2024-04-17 06:16:02 -07:00
trees.cpp Properly handle negative axes in python vmap (#944) 2024-04-02 18:07:23 -07:00
trees.h Properly handle negative axes in python vmap (#944) 2024-04-02 18:07:23 -07:00
utils.h Explicit barriers with concurrent dispatch (#977) 2024-04-10 21:45:31 -07:00