From 9eb4ba8836d9626dfa86d4e9656e25c7782b6d89 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Tue, 30 Mar 2021 16:38:17 +0800 Subject: [PATCH] perf(tensor): always copy to pinned memory before h2d GitOrigin-RevId: f945f2435f5a29a03610a04a47f5d7c1a4562b97 --- imperative/python/src/tensor.cpp | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/imperative/python/src/tensor.cpp b/imperative/python/src/tensor.cpp index a38ee999f..344046f87 100644 --- a/imperative/python/src/tensor.cpp +++ b/imperative/python/src/tensor.cpp @@ -271,10 +271,7 @@ TensorWrapper::TensorWrapper(PyObject* args, PyObject* kwargs) { } interpreter::Interpreter::Handle handle; - constexpr auto size_threshhold = TensorShape::MAX_NDIM; - if (data.size() > size_threshhold) { - handle = interpreter_for_py->put(npy::np2tensor(data.ptr(), npy::Meth::borrow(cn), dtype), no_cache); - } else { + { HostTensorND ret(cn); handle = interpreter_for_py->put(npy::np2tensor(data.ptr(), npy::Meth::copy_into(&ret), dtype), no_cache); } -- GitLab