After we transform our model form ONNX using TensorRT7.1.3 on Jetson AGX, we got the error likes below when inferring:
NVMEDIA_DLA : 495, ERROR: bindOutputTensor failed. err: 0xB
NVMEDIA_DLA : 1920, ERROR: BindOutputTensorArgs failed (Output). status: 0x7.
../rtExt/dla/native/dlaUtils.cpp (194) - DLA Error in submit: 7 (Failure to submit program to DLA engine.)
FAILED_EXECUTION: std::exception
NVMEDIA_DLA : 885, ERROR: runtime registerEvent failed. err: 0x4.
NVMEDIA_DLA : 1849, ERROR: RequestSubmitEvents failed. status: 0x7.
../rtExt/dla/native/dlaUtils.cpp (194) - DLA Error in submit: 7 (Failure to submit program to DLA engine.)
FAILED_EXECUTION: std::exception
...
the struct of our model likes this:
import torch
import torch.nn as nn
import torch.nn.functional as F
import numpy as np
# define network
class Net(nn.Module):
def __init__(self):
super(Net, self).__init__()
self.conv1 = nn.Conv2d(256, 2, 3, 1, 1)
self.conv2 = nn.Conv2d(256, 2, 3, 1, 1)
def forward(self, x):
x = torch.relu(x)
x1 = self.conv1(x)
x2 = self.conv2(x)
x = torch.cat([x1, x2], 1)
return x
net = Net()
a = torch.randn(1,256,40,40)
torch.onnx.export(net, a, "concat.onnx", verbose=True, opset_version=11)
The generated ONNX model is here:
error1.onnx (36.6 KB)