typo

CoinCheung · CoinCheung · commit 78c4d6170533 · 2022-10-07T03:15:50.000Z
diff --git a/tensorrt/CMakeLists.txt b/tensorrt/CMakeLists.txt
@@ -10,8 +10,6 @@ link_directories(/usr/local/cuda/lib64)
 link_directories(${PROJECT_SOURCE_DIR}/build)
 # include_directories(/root/build/TensorRT-8.2.5.1/include)
 # link_directories(/root/build/TensorRT-8.2.5.1/lib)
-include_directories(/data/zzy/trt_install/TensorRT-8.2.5.1/include)
-link_directories(/data/zzy/trt_install/TensorRT-8.2.5.1/lib)
 
 
 find_package(CUDA REQUIRED)
diff --git a/tensorrt/entropy_calibrator.hpp b/tensorrt/entropy_calibrator.hpp
@@ -80,7 +80,7 @@ class EntropyCalibratorImpl
         cudaError_t state;
         state = cudaMemcpy(mDeviceInput, mStream.getBatch(), mInputCount * sizeof(float), cudaMemcpyHostToDevice);
         if (state) {
-            cout << "allocate memory failed\n"; 
+            cout << "memory copy to device failed\n"; 
             std::abort();
         }
         assert(!strcmp(names[0], mInputBlobName));

Original file line number	Diff line number	Diff line change
`@@ -80,7 +80,7 @@ class EntropyCalibratorImpl`
`80`	`80`	`cudaError_t state;`
`81`	`81`	`state = cudaMemcpy(mDeviceInput, mStream.getBatch(), mInputCount * sizeof(float), cudaMemcpyHostToDevice);`
`82`	`82`	`if (state) {`
`83`		`- cout << "allocate memory failed\n";`
	`83`	`+ cout << "memory copy to device failed\n";`
`84`	`84`	`std::abort();`
`85`	`85`	`}`
`86`	`86`	`assert(!strcmp(names[0], mInputBlobName));`