tensorrt-eyelike-oob-poc / run_python_repro.py
Rodion111's picture
Upload folder using huggingface_hub
391ff22 verified
import faulthandler, sys, tensorrt as trt
faulthandler.enable(file=sys.stderr, all_threads=True)
logger = trt.Logger(trt.Logger.VERBOSE)
builder = trt.Builder(logger)
flag = 1 << int(trt.NetworkDefinitionCreationFlag.EXPLICIT_BATCH)
network = builder.create_network(flag)
parser = trt.OnnxParser(network, logger)
with open('eyelike_oob_input.onnx', 'rb') as f:
blob = f.read()
print('about to parse', len(blob), 'bytes', flush=True)
ok = parser.parse(blob)
print('parse() returned', ok)
for i in range(parser.num_errors):
print(' err', i, ':', parser.get_error(i))