Function
free_buffers
(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 |
| 217 | def 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
Tested by
no test coverage detected