We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent be51004 commit 1cf361bCopy full SHA for 1cf361b
2 files changed
Paddle
backends/metax_gpu/CMakeLists.txt
@@ -26,6 +26,20 @@ set(PYTHON_VERSION ${PY_VERSION})
26
27
set(CMAKE_MODULE_PATH "${CMAKE_SOURCE_DIR}/cmake")
28
message(STATUS "CMAKE_MODULE_PATH: ${CMAKE_MODULE_PATH}")
29
+
30
+if(NOT DEFINED PADDLE_WARP_SIZE)
31
+ set(PADDLE_WARP_SIZE 32)
32
+endif()
33
+math(EXPR PADDLE_WARP_MASK "${PADDLE_WARP_SIZE} - 1")
34
+if(PADDLE_WARP_SIZE EQUAL 64)
35
+ set(PADDLE_WARP_SHIFT 6)
36
+else()
37
+ set(PADDLE_WARP_SHIFT 5)
38
39
+add_definitions(-DPADDLE_WARP_SIZE=${PADDLE_WARP_SIZE})
40
+add_definitions(-DPADDLE_WARP_MASK=${PADDLE_WARP_MASK})
41
+add_definitions(-DPADDLE_WARP_SHIFT=${PADDLE_WARP_SHIFT})
42
43
set(WITH_MKLML ON)
44
if(WITH_ARM)
45
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -fPIC")
0 commit comments