diff --git a/src/core/include/megbrain/tensor.h b/src/core/include/megbrain/tensor.h index aae45495..3a613382 100644 --- a/src/core/include/megbrain/tensor.h +++ b/src/core/include/megbrain/tensor.h @@ -548,6 +548,15 @@ class TensorND { ret.reset(storage().proxy_to_default_cpu(), layout()); return ret; } + + template::value>> + HostTensorND proxy_to_comp_node(CompNode cn) const { + HostTensorStorage host_storage; + host_storage.reset(cn, m_storage.size(), m_storage.raw_storage()); + HostTensorND ret; + ret.reset(host_storage, m_layout); + return ret; + } }; /*!