diff --git a/dipu/torch_dipu/csrc_dipu/runtime/core/allocator/DIPUSpinMutex.h b/dipu/torch_dipu/csrc_dipu/runtime/core/allocator/DIPUSpinMutex.h
index c54bec4d5..a2208b883 100644
--- a/dipu/torch_dipu/csrc_dipu/runtime/core/allocator/DIPUSpinMutex.h
+++ b/dipu/torch_dipu/csrc_dipu/runtime/core/allocator/DIPUSpinMutex.h
@@ -2,6 +2,7 @@
 #pragma once
 
 #include <atomic>
+#include <thread>
 
 namespace dipu {