Updates to build libkernel.so under TensorRT 8

2021-08-16 13:59:19 +05:30
parent d2ce966c46
commit eba78e7e78
21 changed files with 1878 additions and 447 deletions
@@ -8,7 +8,7 @@ Int8EntropyCalibrator::Int8EntropyCalibrator(BatchStream& stream, int firstBatch
    mCalibTableFilePath(calibTableFilePath),
    mInputBlobName(inputBlobName.c_str()),
    mReadCache(readCache) {
-    nvinfer1::DimsNCHW dims = mStream.getDims();
+    nvinfer1::Dims4 dims = mStream.getDims();
    mInputCount = mStream.getBatchSize() * dims.c() * dims.h() * dims.w();
    checkCuda(cudaMalloc(&mDeviceInput, mInputCount * sizeof(float)));
    mStream.reset(firstBatch);