(host_arr: np.ndarray, device_ptr: int)
| 228 | |
| 229 | # Wrapper for cudaMemcpy which infers copy size and does error checking |
| 230 | def memcpy_device_to_host(host_arr: np.ndarray, device_ptr: int): |
| 231 | nbytes = host_arr.size * host_arr.itemsize |
| 232 | cuda_call(cudart.cudaMemcpy(host_arr, device_ptr, nbytes, cudart.cudaMemcpyKind.cudaMemcpyDeviceToHost)) |
| 233 | |
| 234 | |
| 235 | def _do_inference_base(inputs, outputs, stream, execute_async): |
nothing calls this directly
no test coverage detected