Skip to content

Commit be6f3f7

Browse files
committed
Demo test_elementwise_pow_op_metax.py is pass!
1 parent 7d9d5f9 commit be6f3f7

6 files changed

Lines changed: 176 additions & 406 deletions

File tree

backends/metax_gpu/CMakeLists.txt

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,8 @@ project(${PROJ_NAME} CXX C CUDA)
1717

1818
set(TARGET_NAME ${PROJ_NAME})
1919

20+
option(WITH_CINN "Compile with CINN support" ON)
21+
2022
find_package(Python3 REQUIRED COMPONENTS Interpreter)
2123
set(PY_VERSION ${Python3_VERSION_MAJOR}.${Python3_VERSION_MINOR})
2224
message(STATUS "Python version detected: ${PY_VERSION}")
@@ -31,6 +33,7 @@ include(paddle)
3133
# 【修改点 1】: 添加 CINN 子目录编译
3234
if(WITH_CINN)
3335
message(STATUS "[MetaX] CINN enabled, adding subdirectory: cinn")
36+
add_definitions(-DWITH_CINN)
3437
add_subdirectory(cinn)
3538
endif()
3639

0 commit comments

Comments
 (0)