[XPU]. Fixed the bug in buffer free to wait stream.

PaddlePaddle · Jan 31, 2023 · f895534 · f895534
1 parent faa6135
commit f895534
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/lite/backends/xpu/xpu_buffer.h b/lite/backends/xpu/xpu_buffer.h
@@ -29,15 +29,16 @@ class XPUBuffer : public Buffer {
   XPUBuffer() {
     target_ = TargetType::kXPU;
     xpu_l3_cache_block_ = lite::TargetWrapperXPU::CreateL3CacheBlock();
+    xpu_stream_ = TargetWrapperXPU::get_xpu_stream();
   }
 
   void ResetLazy(TargetType target, size_t size) override {
     CHECK(target == TargetType::kXPU);
     CHECK_EQ(own_data_, true) << "Can not reset unowned buffer.";
     if (global_mem_space_ < size) {
       if (global_mem_space_ > 0) {
-        void* xpu_stream = TargetWrapperXPU::get_xpu_stream();
-        XPU_CALL(xpu_wait(xpu_stream));
+        xpu_stream_ = TargetWrapperXPU::get_xpu_stream();
+        XPU_CALL(xpu_wait(xpu_stream_));
         XPU_CALL(xpu_free(global_mem_data_));
         global_mem_data_ = nullptr;
       }