bugfix in unittest

Author: mag1c-h

ucm/store/cache/cc/dump_queue.cc

@@ -136,14 +136,13 @@ Status DumpQueue::DumpOneTask(Trans::Stream* stream, size_t tensorSize, TaskPtr
 void DumpQueue::BackendDumpStage()
 {
     dumping_.ConsumerLoop(stop_, [this](auto&& task) {
-        static Detail::TaskHandle finishedBackendTaskHandle = 0;
-        if (task.backendTaskHandle > finishedBackendTaskHandle) {
+        if (task.backendTaskHandle > finishedBackendTaskHandle_) {
             auto s = backend_->Wait(task.backendTaskHandle);
             if (s.Failure()) {
                 UC_ERROR("Failed({}) to wait backend task({}).", s, task.backendTaskHandle);
                 return;
             }
-            finishedBackendTaskHandle = task.backendTaskHandle;
+            finishedBackendTaskHandle_ = task.backendTaskHandle;
         }
     });
 }

ucm/store/cache/cc/dump_queue.h

@@ -48,6 +48,7 @@ class DumpQueue {
 
 private:
     alignas(64) std::atomic_bool stop_{false};
+    Detail::TaskHandle finishedBackendTaskHandle_{0};
     TaskIdSet* failureSet_{nullptr};
     TransBuffer* buffer_{nullptr};
     Store* backend_{nullptr};

ucm/store/cache/cc/load_queue.cc

@@ -121,8 +121,7 @@ void LoadQueue::TransferStage(int32_t deviceId, size_t tensorSize, std::promise<
 
 void LoadQueue::TransferOneTask(Trans::Stream* stream, size_t tensorSize, ShardTask&& task)
 {
-    static Detail::TaskHandle finishedBackendTaskHandle = 0;
-    if (task.backendTaskHandle > finishedBackendTaskHandle) {
+    if (task.backendTaskHandle > finishedBackendTaskHandle_) {
         auto s = backend_->Wait(task.backendTaskHandle);
         if (s.Failure()) [[unlikely]] {
             failureSet_->Insert(task.taskHandle);
@@ -131,7 +130,7 @@ void LoadQueue::TransferOneTask(Trans::Stream* stream, size_t tensorSize, ShardT
             return;
         }
         task.bufferHandle.MarkReady();
-        finishedBackendTaskHandle = task.backendTaskHandle;
+        finishedBackendTaskHandle_ = task.backendTaskHandle;
     }
     for (;;) {
         if (failureSet_->Contains(task.taskHandle)) {

ucm/store/cache/cc/load_queue.h

@@ -51,6 +51,7 @@ class LoadQueue {
 
 private:
     alignas(64) std::atomic_bool stop_{false};
+    Detail::TaskHandle finishedBackendTaskHandle_{0};
     TaskIdSet* failureSet_{nullptr};
     TransBuffer* buffer_{nullptr};
     Store* backend_{nullptr};