zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-07-02 23:31:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Remove Event::Signal() (#2052)

Browse Source
This commit is contained in:
Cheng 2025-04-08 22:20:27 +09:00 committed by GitHub
parent 60c4154346
commit 08a1bf3f10
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 11 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mlx/backend/metal/event.cpp
View File

@ -24,10 +24,6 @@ void Event::wait() {
}
}
void Event::signal() {
static_cast<MTL::SharedEvent*>(event_.get())->setSignaledValue(value());
}
void Event::wait(Stream stream) {
if (stream.device == Device::cpu) {
scheduler::enqueue(stream, [*this]() mutable { wait(); });
@ -42,7 +38,9 @@ void Event::wait(Stream stream) {
void Event::signal(Stream stream) {
if (stream.device == Device::cpu) {
scheduler::enqueue(stream, [*this]() mutable { signal(); });
scheduler::enqueue(stream, [*this]() mutable {
static_cast<MTL::SharedEvent*>(event_.get())->setSignaledValue(value());
});
} else {
auto& d = metal::device(stream.device);
d.end_encoding(stream.index);

18
mlx/backend/no_metal/event.cpp
View File

@ -28,21 +28,19 @@ void Event::wait() {
ec->cv.wait(lk, [value = value(), ec] { return ec->value >= value; });
}
void Event::signal() {
auto ec = static_cast<EventCounter*>(event_.get());
{
std::lock_guard<std::mutex> lk(ec->mtx);
ec->value = value();
}
ec->cv.notify_all();
}
void Event::wait(Stream stream) {
scheduler::enqueue(stream, [*this]() mutable { wait(); });
}
void Event::signal(Stream stream) {
scheduler::enqueue(stream, [*this]() mutable { signal(); });
scheduler::enqueue(stream, [*this]() mutable {
auto ec = static_cast<EventCounter*>(event_.get());
{
std::lock_guard<std::mutex> lk(ec->mtx);
ec->value = value();
}
ec->cv.notify_all();
});
}
bool Event::is_signaled() const {

3
mlx/event.h
View File

@ -16,9 +16,6 @@ class Event {
// Wait for the event to be signaled at its current value
void wait();
// Signal the event at its current value
void signal();
// Wait in the given stream for the event to be signaled at its current value
void wait(Stream stream);