Skip to content

Commit 3c83d18

Browse files
felixhao28ZeyuChen
andauthored
Fix compiling faster transformer with cuda 11 (PaddlePaddle#1234)
Co-authored-by: Zeyu Chen <[email protected]>
1 parent af6e432 commit 3c83d18

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

paddlenlp/ops/faster_transformer/src/CMakeLists.txt

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,8 @@
1+
if (${CUDA_VERSION} GREATER_EQUAL 11.0)
2+
message(STATUS "Add DCUDA11_MODE")
3+
add_definitions("-DCUDA11_MODE")
4+
endif()
5+
16
add_definitions(-DNDEBUG)
27
add_definitions(-DPADDLE_CUDA)
38
# Default is 1 in standard c++ when using gcc8.2

0 commit comments

Comments
 (0)