File tree Expand file tree Collapse file tree 4 files changed +6
-4
lines changed
python/paddle/fluid/tests/unittests Expand file tree Collapse file tree 4 files changed +6
-4
lines changed Original file line number Diff line number Diff line change 4
4
5
5
6
6
if (WITH_NV_JETSON )
7
+ add_definitions (-DWITH_NV_JETSON )
7
8
set (paddle_known_gpu_archs "53 62 72" )
8
9
set (paddle_known_gpu_archs7 "53" )
9
10
set (paddle_known_gpu_archs8 "53 62" )
Original file line number Diff line number Diff line change @@ -284,11 +284,12 @@ function(version version_file)
284
284
if (WITH_GPU )
285
285
file (APPEND ${version_file}
286
286
"CUDA version: ${CUDA_VERSION} \n "
287
- "CUDNN version: v${CUDNN_MAJOR_VERSION} \n " )
287
+ "CUDNN version: v${CUDNN_MAJOR_VERSION} . ${CUDNN_MINOR_VERSION} \n " )
288
288
endif ()
289
+ file (APPEND ${version_file} "CXX compiler version: ${CMAKE_CXX_COMPILER_VERSION} \n " )
289
290
if (TENSORRT_FOUND )
290
291
file (APPEND ${version_file}
291
- "WITH_TENSORRT: ${TENSORRT_FOUND} \n " )
292
+ "WITH_TENSORRT: ${TENSORRT_FOUND} \n " "TensorRT version: v ${TENSORRT_MAJOR_VERSION} \n " )
292
293
endif ()
293
294
294
295
endfunction ()
Original file line number Diff line number Diff line change @@ -53,7 +53,7 @@ if (NOT WITH_MKL OR NOT WITH_AVX)
53
53
SET (OP_MKL_DEPS ${OP_MKL_DEPS} match_matrix_tensor_op )
54
54
SET (OP_MKL_DEPS ${OP_MKL_DEPS} var_conv_2d_op )
55
55
endif ()
56
- if (WITH_COVERAGE OR NOT WITH_AVX OR WIN32 )
56
+ if (WITH_COVERAGE OR NOT WITH_AVX OR WIN32 OR WITH_NV_JETSON )
57
57
SET (OP_MKL_DEPS ${OP_MKL_DEPS} pyramid_hash_op )
58
58
endif ()
59
59
Original file line number Diff line number Diff line change @@ -91,7 +91,7 @@ if(NOT WITH_MKL OR NOT WITH_AVX)
91
91
list (REMOVE_ITEM TEST_OPS test_match_matrix_tensor_op )
92
92
list (REMOVE_ITEM TEST_OPS test_var_conv_2d )
93
93
endif ()
94
- if (WITH_COVERAGE OR NOT WITH_AVX OR WIN32 )
94
+ if (WITH_COVERAGE OR NOT WITH_AVX OR WIN32 OR WITH_NV_JETSON )
95
95
list (REMOVE_ITEM TEST_OPS test_pyramid_hash_op )
96
96
list (REMOVE_ITEM TEST_OPS test_fleet_pyramid_hash )
97
97
endif ()
You can’t perform that action at this time.
0 commit comments