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

Function cuda_call

samples/python/common.py:45–50  ·  view source on GitHub ↗
(call)

Source from the content-addressed store, hash-verified

43 raise RuntimeError("Unknown error type: {}".format(err))
44
45def cuda_call(call):
46 err, res = call[0], call[1:]
47 check_cuda_err(err)
48 if len(res) == 1:
49 res = res[0]
50 return res
51
52def GiB(val):
53 return val * 1 << 30

Callers 7

__init__Method · 0.85
freeMethod · 0.85
allocate_buffersFunction · 0.85
free_buffersFunction · 0.85
memcpy_host_to_deviceFunction · 0.85
memcpy_device_to_hostFunction · 0.85
_do_inference_baseFunction · 0.85

Calls 1

check_cuda_errFunction · 0.85

Tested by

no test coverage detected