(tvm) branch main updated: Fix includes of custom allreduce kernel (#16814)

[tqchen]

This is an automated email from the ASF dual-hosted git repository.

tqchen pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new 3ce87cba21 Fix includes of custom allreduce kernel (#16814)
3ce87cba21 is described below

commit 3ce87cba21cf1cccf45a8f3ad57e94e05db51b0d
Author: Wuwei Lin <wu...@apache.org>
AuthorDate: Fri Mar 29 03:59:28 2024 -0700

    Fix includes of custom allreduce kernel (#16814)
---
 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 6dec368b43..36ac0e3a43 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"