From 98134777d9fc55e42668a664ca37310d3f28647c Mon Sep 17 00:00:00 2001
From: Wuwei Lin <wuwei@apache.org>
Date: Thu, 28 Mar 2024 15:28:50 -0700
Subject: [PATCH] Fix includes of custom allreduce kernel

---
 3rdparty/tensorrt_llm/custom_allreduce_kernels.cu | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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"