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

Function free_buffers

samples/python/common.py:217–220  ·  view source on GitHub ↗
(inputs: List[HostDeviceMem], outputs: List[HostDeviceMem], stream: cudart.cudaStream_t)

Source from the content-addressed store, hash-verified

215
216# Frees the resources allocated in allocate_buffers
217def free_buffers(inputs: List[HostDeviceMem], outputs: List[HostDeviceMem], stream: cudart.cudaStream_t):
218 for mem in inputs + outputs:
219 mem.free()
220 cuda_call(cudart.cudaStreamDestroy(stream))
221
222
223# Wrapper for cudaMemcpy which infers copy size and does error checking

Callers

nothing calls this directly

Calls 2

cuda_callFunction · 0.85
freeMethod · 0.45

Tested by

no test coverage detected