|
1 | 1 | cmake_minimum_required(VERSION 3.18) |
2 | | -project(InfiniOps VERSION 0.1.0 LANGUAGES CXX) |
3 | | - |
4 | | -include(GNUInstallDirs) |
| 2 | +project(InfiniOps LANGUAGES CXX) |
5 | 3 |
|
6 | 4 | if(POLICY CMP0116) |
7 | 5 | cmake_policy(SET CMP0116 NEW) |
@@ -34,7 +32,6 @@ option(BUILD_CUSTOM_KERNEL "Build custom AscendC kernel PyTorch extension (requi |
34 | 32 |
|
35 | 33 | option(AUTO_DETECT_DEVICES "Automatically detect available devices" OFF) |
36 | 34 | option(AUTO_DETECT_BACKENDS "Automatically detect available backends" OFF) |
37 | | -option(GENERATE_CPP_OPERATOR_API "Generate public C++ operator API" ON) |
38 | 35 | option(GENERATE_PYTHON_BINDINGS "Generate Python bindings" OFF) |
39 | 36 |
|
40 | 37 | set(_DEFAULT_HYGON_DTK_ROOT "/opt/dtk") |
@@ -312,13 +309,10 @@ if(WITH_ILUVATAR) |
312 | 309 | if(NOT ILUVATAR_CUDA_COMPILER) |
313 | 310 | message(FATAL_ERROR "`WITH_ILUVATAR` is `ON` but CoreX `clang++` was not found.") |
314 | 311 | endif() |
315 | | - get_filename_component(ILUVATAR_CUDA_BIN_DIR "${ILUVATAR_CUDA_COMPILER}" DIRECTORY) |
316 | | - get_filename_component(ILUVATAR_CUDA_ROOT "${ILUVATAR_CUDA_BIN_DIR}/.." ABSOLUTE) |
317 | | - set(CUDAToolkit_ROOT "${ILUVATAR_CUDA_ROOT}" CACHE PATH "Iluvatar CoreX toolkit root") |
318 | 312 | set(ILUVATAR_CUDA_FLAGS |
319 | | - "--cuda-gpu-arch=${ILUVATAR_ARCH};-fPIC;-Wno-error=unused-variable;-Wno-error=unused-private-field;-Wno-unused-variable;-std=c++17;--cuda-path=${ILUVATAR_CUDA_ROOT};-x;ivcore" |
| 313 | + "--cuda-gpu-arch=${ILUVATAR_ARCH};-fPIC;-Wno-error=unused-variable;-Wno-error=unused-private-field;-Wno-unused-variable;-std=c++17;--cuda-path=/usr/local/corex;-x;ivcore" |
320 | 314 | CACHE STRING "Iluvatar CUDA compiler flags") |
321 | | - message(STATUS "Iluvatar: CUDA compiler ${ILUVATAR_CUDA_COMPILER}, arch ${ILUVATAR_ARCH}, toolkit ${ILUVATAR_CUDA_ROOT}") |
| 315 | + message(STATUS "Iluvatar: CUDA compiler ${ILUVATAR_CUDA_COMPILER}, arch ${ILUVATAR_ARCH}") |
322 | 316 | find_package(CUDAToolkit REQUIRED) |
323 | 317 | endif() |
324 | 318 |
|
@@ -436,7 +430,6 @@ if(WITH_MOORE) |
436 | 430 | find_library(MUSA_LIB NAMES musa HINTS "${MUSA_ROOT}/lib" REQUIRED) |
437 | 431 | find_library(MUSART_LIB NAMES musart HINTS "${MUSA_ROOT}/lib" REQUIRED) |
438 | 432 | find_library(MUBLAS_LIB NAMES mublas HINTS "${MUSA_ROOT}/lib" REQUIRED) |
439 | | - find_library(MUSA_OPENMP_LIB NAMES omp iomp5 HINTS "${MUSA_ROOT}/lib" REQUIRED) |
440 | 433 | endif() |
441 | 434 |
|
442 | 435 | if(WITH_CAMBRICON) |
@@ -471,7 +464,6 @@ endif() |
471 | 464 |
|
472 | 465 | # If all other platforms are not enabled, CPU is enabled by default. |
473 | 466 | if(NOT WITH_NVIDIA AND NOT WITH_ILUVATAR AND NOT WITH_HYGON AND NOT WITH_METAX AND NOT WITH_MOORE AND NOT WITH_CAMBRICON AND NOT WITH_ASCEND) |
474 | | - set(WITH_CPU ON CACHE BOOL "Enable CPU backend" FORCE) |
475 | 467 | add_compile_definitions(WITH_CPU=1) |
476 | 468 | endif() |
477 | 469 |
|
|
0 commit comments