This repository was archived by the owner on Apr 28, 2023. It is now read-only.
File tree 3 files changed +6
-6
lines changed
3 files changed +6
-6
lines changed Original file line number Diff line number Diff line change @@ -315,8 +315,8 @@ if (WITH_CAFFE2 AND WITH_CUDA)
315
315
message (STATUS "Found, CAFFE2_INCLUDE_DIR: " ${CAFFE2_INCLUDE_DIR} )
316
316
317
317
add_definitions (-DCAFFE2_USE_GOOGLE_GLOG)
318
- include_directories (AFTER ${EIGEN_INCLUDE_DIR} )
319
- include_directories (AFTER ${CAFFE2_INCLUDE_DIR} )
318
+ include_directories (AFTER SYSTEM ${EIGEN_INCLUDE_DIR} )
319
+ include_directories (AFTER SYSTEM ${CAFFE2_INCLUDE_DIR} )
320
320
add_subdirectory (tc/c2)
321
321
add_subdirectory (tc/benchmarks)
322
322
add_subdirectory (test /caffe2)
Original file line number Diff line number Diff line change @@ -286,8 +286,8 @@ void ProductionModel::runATen1LUT() {
286
286
}
287
287
288
288
void ProductionModel::run2LUT (const tc::CudaMappingOptions& options) {
289
- TC_CHECK_LT (0 , E1 );
290
- TC_CHECK_LT (0 , E2 );
289
+ TC_CHECK_LT (0u , E1 );
290
+ TC_CHECK_LT (0u , E2 );
291
291
auto ws_init_func = [=](Workspace& w) {
292
292
AddDeterministicallyRandomInput<caffe2::CUDABackend, float >(
293
293
w, {E1 , D}, " LUT1" );
Original file line number Diff line number Diff line change @@ -65,11 +65,11 @@ struct Benchmark : public ::testing::Test {
65
65
void SetUp () {
66
66
if (!FLAGS_disable_version_checks) {
67
67
auto cudnnVersion = cudnnGetVersion ();
68
- TC_CHECK_LE (6021 , cudnnVersion)
68
+ TC_CHECK_LE (6021u , cudnnVersion)
69
69
<< " [CUDNN][VERSION] Enforce version compatibility check" ;
70
70
71
71
auto cudaRtVersion = cudnnGetCudartVersion ();
72
- TC_CHECK_LE (8000 , cudaRtVersion)
72
+ TC_CHECK_LE (8000u , cudaRtVersion)
73
73
<< " [CUDART][VERSION] Enforce version compatibility check" ;
74
74
75
75
int cublasVersion;
You can’t perform that action at this time.
0 commit comments