Skip to content

Commit 1cf361b

Browse files
committed
Update Paddle.
1 parent be51004 commit 1cf361b

2 files changed

Lines changed: 15 additions & 1 deletion

File tree

Paddle

Submodule Paddle updated 1296 files

backends/metax_gpu/CMakeLists.txt

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,20 @@ set(PYTHON_VERSION ${PY_VERSION})
2626

2727
set(CMAKE_MODULE_PATH "${CMAKE_SOURCE_DIR}/cmake")
2828
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+
endif()
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+
2943
set(WITH_MKLML ON)
3044
if(WITH_ARM)
3145
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} -fPIC")

0 commit comments

Comments
 (0)