[tvm] branch main updated: fix device on HandleCopyFromRemote (#9616)

[ar...@apache.org]

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

areusch 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 a76f648  fix device on HandleCopyFromRemote (#9616)
a76f648 is described below

commit a76f648b3ed2a67a7afc62bea6ea65b139015bff
Author: Mehrdad Hessar <mh...@octoml.ai>
AuthorDate: Wed Dec 1 08:48:28 2021 -0800

    fix device on HandleCopyFromRemote (#9616)
---
 src/runtime/minrpc/minrpc_server.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/runtime/minrpc/minrpc_server.h b/src/runtime/minrpc/minrpc_server.h
index 1dfee70..64925a2 100644
--- a/src/runtime/minrpc/minrpc_server.h
+++ b/src/runtime/minrpc/minrpc_server.h
@@ -192,7 +192,7 @@ class MinRPCServer {
       data_ptr = this->ArenaAlloc<uint8_t>(num_bytes);
       DLTensor temp;
       temp.data = reinterpret_cast<void*>(data_ptr);
-      temp.device = arr->device;
+      temp.device = DLDevice{kDLCPU, 0};
       temp.ndim = arr->ndim;
       temp.dtype = arr->dtype;
       temp.shape = arr->shape;