(tvm) branch main updated: [Runtime] Add "TVM_DLL" to NDArray cache load func (#16541)

[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 32f83d581c [Runtime] Add "TVM_DLL" to NDArray cache load func (#16541)
32f83d581c is described below

commit 32f83d581c71e85ddc3aca06da3c9c1fa02deae0
Author: Ruihang Lai <ruiha...@cs.cmu.edu>
AuthorDate: Mon Feb 12 12:35:58 2024 -0500

    [Runtime] Add "TVM_DLL" to NDArray cache load func (#16541)
    
    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 584da8f0ca..d8cc155824 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<NDArray>* staging_buffer = nullptr) const;
+      TVM_DLL NDArray Load(Device device, const std::string* raw_data,
+                           Optional<NDArray>* staging_buffer = nullptr) const;
 
       /*! \brief Name of the parameter */
       std::string name;