diff --git a/3rdparty/tensorrt_llm/custom_allreduce_kernels.cu b/3rdparty/tensorrt_llm/custom_allreduce_kernels.cu
index 6dec368b4380..36ac0e3a439e 100644
--- a/3rdparty/tensorrt_llm/custom_allreduce_kernels.cu
+++ b/3rdparty/tensorrt_llm/custom_allreduce_kernels.cu
@@ -15,8 +15,9 @@
  */
 
 #include <cuda_fp16.h>
-#include <dmlc/logging.h>
+#include <dlpack/dlpack.h>
 #include <stdint.h>
+#include <tvm/runtime/logging.h>
 
 #include "custom_allreduce_kernels.h"