Revert "Updates to build libkernel.so under TensorRT 8"

This reverts commit eba78e7e78.
This commit is contained in:
perseusdg
2021-08-16 14:07:07 +05:30
parent 52db97a140
commit 19b1d6d6f7
21 changed files with 447 additions and 1878 deletions
+1 -1
View File
@@ -8,7 +8,7 @@ Int8EntropyCalibrator::Int8EntropyCalibrator(BatchStream& stream, int firstBatch
mCalibTableFilePath(calibTableFilePath),
mInputBlobName(inputBlobName.c_str()),
mReadCache(readCache) {
nvinfer1::Dims4 dims = mStream.getDims();
nvinfer1::DimsNCHW dims = mStream.getDims();
mInputCount = mStream.getBatchSize() * dims.c() * dims.h() * dims.w();
checkCuda(cudaMalloc(&mDeviceInput, mInputCount * sizeof(float)));
mStream.reset(firstBatch);