Faster synchronization Fence primitive (#1773)

* try faster synchronization

move event

fixes

update bench

fix

fix

* non-functioning kernel

* try alternative fence

* cleanup barrier

* get rid of event_fence

* update benchmarks

* doc string in metal fence
This commit is contained in:
Awni Hannun
2025-01-17 18:42:19 -08:00
committed by GitHub
parent 0c259961ac
commit a4667da1eb
11 changed files with 362 additions and 31 deletions

View File

@@ -126,6 +126,11 @@ inline int max_ops_per_buffer() {
return max_ops_per_buffer_;
}
inline bool metal_fast_synch() {
static bool metal_fast_synch = get_var("MLX_METAL_FAST_SYNCH", 0);
return metal_fast_synch;
}
} // namespace env
} // namespace mlx::core