From 2be70c4178abb7c56f250d5622656dfceafbe414 Mon Sep 17 00:00:00 2001
From: wathenjiang <wathenjiang@tencent.com>
Date: Wed, 20 Sep 2023 13:48:54 +0800
Subject: [PATCH] use atomic type in loom instead

---
 tokio/src/runtime/task/list.rs | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/tokio/src/runtime/task/list.rs b/tokio/src/runtime/task/list.rs
index ff109a4a3e4..293ec4d9781 100644
--- a/tokio/src/runtime/task/list.rs
+++ b/tokio/src/runtime/task/list.rs
@@ -13,10 +13,9 @@ use crate::loom::sync::Mutex;
 use crate::runtime::task::{JoinHandle, LocalNotified, Notified, Schedule, Task};
 use crate::util::linked_list::{Link, LinkedList};
 
+use crate::loom::sync::atomic::{AtomicBool, Ordering};
 use std::marker::PhantomData;
 use std::num::NonZeroU32;
-use std::sync::atomic::AtomicBool;
-use std::sync::atomic::{AtomicU32, Ordering};
 
 // The id from the module below is used to verify whether a given task is stored
 // in this OwnedTasks, or some other task. The counter starts at one so we can
@@ -27,7 +26,7 @@ use std::sync::atomic::{AtomicU32, Ordering};
 // bug in Tokio, so we accept that certain bugs would not be caught if the two
 // mixed up runtimes happen to have the same id.
 
-static NEXT_OWNED_TASKS_ID: AtomicU32 = AtomicU32::new(1);
+static NEXT_OWNED_TASKS_ID: std::sync::atomic::AtomicU32 = std::sync::atomic::AtomicU32::new(1);
 
 fn get_next_id() -> NonZeroU32 {
     loop {