|
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) |
@@ -36,7 +34,6 @@ option(BUILD_CUSTOM_KERNEL "Build custom AscendC kernel PyTorch extension (requi |
36 | 34 |
|
37 | 35 | option(AUTO_DETECT_DEVICES "Automatically detect available devices" OFF) |
38 | 36 | option(AUTO_DETECT_BACKENDS "Automatically detect available backends" OFF) |
39 | | -option(GENERATE_CPP_OPERATOR_API "Generate public C++ operator API" ON) |
40 | 37 | option(GENERATE_PYTHON_BINDINGS "Generate Python bindings" OFF) |
41 | 38 |
|
42 | 39 | set(_DEFAULT_HYGON_DTK_ROOT "/opt/dtk") |
@@ -322,13 +319,10 @@ if(WITH_ILUVATAR) |
322 | 319 | if(NOT ILUVATAR_CUDA_COMPILER) |
323 | 320 | message(FATAL_ERROR "`WITH_ILUVATAR` is `ON` but CoreX `clang++` was not found.") |
324 | 321 | endif() |
325 | | - get_filename_component(ILUVATAR_CUDA_BIN_DIR "${ILUVATAR_CUDA_COMPILER}" DIRECTORY) |
326 | | - get_filename_component(ILUVATAR_CUDA_ROOT "${ILUVATAR_CUDA_BIN_DIR}/.." ABSOLUTE) |
327 | | - set(CUDAToolkit_ROOT "${ILUVATAR_CUDA_ROOT}" CACHE PATH "Iluvatar CoreX toolkit root") |
328 | 322 | set(ILUVATAR_CUDA_FLAGS |
329 | | - "--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" |
| 323 | + "--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" |
330 | 324 | CACHE STRING "Iluvatar CUDA compiler flags") |
331 | | - message(STATUS "Iluvatar: CUDA compiler ${ILUVATAR_CUDA_COMPILER}, arch ${ILUVATAR_ARCH}, toolkit ${ILUVATAR_CUDA_ROOT}") |
| 325 | + message(STATUS "Iluvatar: CUDA compiler ${ILUVATAR_CUDA_COMPILER}, arch ${ILUVATAR_ARCH}") |
332 | 326 | find_package(CUDAToolkit REQUIRED) |
333 | 327 | endif() |
334 | 328 |
|
@@ -446,7 +440,6 @@ if(WITH_MOORE) |
446 | 440 | find_library(MUSA_LIB NAMES musa HINTS "${MUSA_ROOT}/lib" REQUIRED) |
447 | 441 | find_library(MUSART_LIB NAMES musart HINTS "${MUSA_ROOT}/lib" REQUIRED) |
448 | 442 | find_library(MUBLAS_LIB NAMES mublas HINTS "${MUSA_ROOT}/lib" REQUIRED) |
449 | | - find_library(MUSA_OPENMP_LIB NAMES omp iomp5 HINTS "${MUSA_ROOT}/lib" REQUIRED) |
450 | 443 | endif() |
451 | 444 |
|
452 | 445 | if(WITH_CAMBRICON) |
@@ -481,7 +474,6 @@ endif() |
481 | 474 |
|
482 | 475 | # If all other platforms are not enabled, CPU is enabled by default. |
483 | 476 | 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) |
484 | | - set(WITH_CPU ON CACHE BOOL "Enable CPU backend" FORCE) |
485 | 477 | add_compile_definitions(WITH_CPU=1) |
486 | 478 | endif() |
487 | 479 |
|
|
0 commit comments