diff --git a/mlx/backend/cuda/event.cu b/mlx/backend/cuda/event.cu index afa032a83..f51d2f2e3 100644 --- a/mlx/backend/cuda/event.cu +++ b/mlx/backend/cuda/event.cu @@ -90,8 +90,6 @@ bool CudaEvent::completed() const { // SharedEvent implementations /////////////////////////////////////////////////////////////////////////////// -namespace { - __host__ __device__ void event_wait(SharedEvent::Atomic* ac, uint64_t value) { uint64_t current; while ((current = ac->load()) < value) { @@ -112,8 +110,6 @@ __global__ void event_signal_kernel(SharedEvent::Atomic* ac, uint64_t value) { event_signal(ac, value); } -} // namespace - SharedEvent::SharedEvent() { // Allocate cuda::atomic on managed memory. Atomic* ac;