diff --git a/src/ray/core_worker/context.cc b/src/ray/core_worker/context.cc index a19d03f537d6b..d931851f40379 100644 --- a/src/ray/core_worker/context.cc +++ b/src/ray/core_worker/context.cc @@ -395,7 +395,7 @@ bool WorkerContext::CurrentActorIsAsync() const { return current_actor_is_asyncio_; } -void WorkerContext::SetCurrentActorShouldExit() ABSL_LOCKS_EXCLUDED(mutex_) { +void WorkerContext::SetCurrentActorShouldExit() { absl::WriterMutexLock lock(&mutex_); RAY_CHECK(!current_actor_id_.IsNil()); current_actor_should_exit_ = true;