From 50382cc6b21ff032e652cf3dfea28adb5b2a925c Mon Sep 17 00:00:00 2001 From: Ruihang Lai Date: Wed, 7 Feb 2024 14:31:24 -0500 Subject: [PATCH] [Runtime] Add "TVM_DLL" to NDArray cache load func The `NDArrayCacheMetadata::Load` function lacks the `TVM_DLL` attribute which leads to build failure on Windows. This PR fixes the issue. --- include/tvm/runtime/relax_vm/ndarray_cache_support.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/include/tvm/runtime/relax_vm/ndarray_cache_support.h b/include/tvm/runtime/relax_vm/ndarray_cache_support.h index 584da8f0ca81..d8cc155824d5 100644 --- a/include/tvm/runtime/relax_vm/ndarray_cache_support.h +++ b/include/tvm/runtime/relax_vm/ndarray_cache_support.h @@ -45,8 +45,8 @@ struct NDArrayCacheMetadata { * \param staging_buffer The buffer to be used to avoid extra OpenCL copies. Pass in a nullptr * in other cases */ - NDArray Load(Device device, const std::string* raw_data, - Optional* staging_buffer = nullptr) const; + TVM_DLL NDArray Load(Device device, const std::string* raw_data, + Optional* staging_buffer = nullptr) const; /*! \brief Name of the parameter */ std::string name;