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

Function make_cuda_shared

plugin/common/bertCommon.h:309–312  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

307
308template <typename T>
309void make_cuda_shared(cuda_shared_ptr<T>& ptr, void* cudaMem)
310{
311 ptr.reset(static_cast<T*>(cudaMem), bert::CudaDeleter<T>());
312}
313
314struct WeightsWithOwnership : public nvinfer1::Weights
315{

Callers 1

GeluPluginDynamicMethod · 0.85

Calls 1

resetMethod · 0.45

Tested by

no test coverage detected