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