MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT / serFromDev

Function serFromDev

plugin/common/bertCommon.h:181–186  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

179
180template <typename T>
181inline void serFromDev(char*& buffer, T const* data, size_t nbElem)
182{
183 const size_t len = sizeof(T) * nbElem;
184 PLUGIN_CUASSERT(cudaMemcpy(buffer, static_cast<void const*>(data), len, cudaMemcpyDeviceToHost));
185 buffer += len;
186}
187
188template <typename T>
189inline T* devToDev(T const* data, size_t nbElem)

Callers 6

serializeMethod · 0.85
serializeMethod · 0.85
serializeMethod · 0.85
serializeMethod · 0.85
serializeMethod · 0.85
serializeMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected