(tvm) branch refactor-s2 updated: Update to fix gpu and ffi

[tqchen]

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

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


The following commit(s) were added to refs/heads/refactor-s2 by this push:
     new 46d8854ace Update to fix gpu and ffi
46d8854ace is described below

commit 46d8854acea671513303cd5375a45589797c75c3
Author: tqchen <tianqi.tc...@gmail.com>
AuthorDate: Mon Apr 28 20:51:07 2025 -0400

    Update to fix gpu and ffi
---
 3rdparty/cutlass_fpA_intB_gemm          | 2 +-
 ffi/include/tvm/ffi/container/ndarray.h | 6 ++++--
 ffi/include/tvm/ffi/function.h          | 1 +
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/3rdparty/cutlass_fpA_intB_gemm b/3rdparty/cutlass_fpA_intB_gemm
index fdef230791..d463617c21 160000
--- a/3rdparty/cutlass_fpA_intB_gemm
+++ b/3rdparty/cutlass_fpA_intB_gemm
@@ -1 +1 @@
-Subproject commit fdef2307917ec2c7cc5becc29fb95d77498484bd
+Subproject commit d463617c215969d9d2333a216042ddb963949d0d
diff --git a/ffi/include/tvm/ffi/container/ndarray.h 
b/ffi/include/tvm/ffi/container/ndarray.h
index 66bdbb6d70..065c8db2cd 100644
--- a/ffi/include/tvm/ffi/container/ndarray.h
+++ b/ffi/include/tvm/ffi/container/ndarray.h
@@ -29,6 +29,8 @@
 #include <tvm/ffi/error.h>
 #include <tvm/ffi/type_traits.h>
 
+#include <utility>
+
 namespace tvm {
 namespace ffi {
 
@@ -85,7 +87,7 @@ inline bool IsAligned(const DLTensor& arr, size_t alignment) {
  */
 inline size_t GetDataSize(int64_t numel, DLDataType dtype) {
   // compatible handling sub-byte uint1(bool), which usually stored as uint8_t
-  // TODO: revisit and switch to kDLBool
+  // TODO(tqchen): revisit and switch to kDLBool
   if (dtype.code == kDLUInt && dtype.bits == 1 && dtype.lanes == 1) {
     return numel;
   }
@@ -197,7 +199,7 @@ class NDArrayObjFromNDAlloc : public NDArrayObj {
 template <typename TDLPackManagedTensor>
 class NDArrayObjFromDLPack : public NDArrayObj {
  public:
-  NDArrayObjFromDLPack(TDLPackManagedTensor* tensor) : tensor_(tensor) {
+  explicit NDArrayObjFromDLPack(TDLPackManagedTensor* tensor) : 
tensor_(tensor) {
     *static_cast<DLTensor*>(this) = tensor_->dl_tensor;
     // set strides to nullptr if the tensor is contiguous.
     if (IsContiguous(tensor->dl_tensor)) {
diff --git a/ffi/include/tvm/ffi/function.h b/ffi/include/tvm/ffi/function.h
index d53bb01934..d5e31372c6 100644
--- a/ffi/include/tvm/ffi/function.h
+++ b/ffi/include/tvm/ffi/function.h
@@ -32,6 +32,7 @@
 #include <functional>
 #include <string>
 #include <utility>
+#include <vector>
 
 namespace tvm {
 namespace ffi {