You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

CMakeLists.txt 666 B

11 months ago
123456789101112131415
  1. cmake_minimum_required(VERSION 3.19)
  2. project(uctc)
  3. set(CMAKE_CXX_STANDARD 17)
  4. set(CXX g++)
  5. set(CMAKE_CXX_FLAGS -O3)
  6. set(PYBIND11_DIR /home/hce/miniconda3/lib/python3.11/site-packages/pybind11/share/cmake/pybind11)
  7. set(PYBIND11_FINDPYTHON ON)
  8. find_package(pybind11 CONFIG REQUIRED PATHS ${PYBIND11_DIR})
  9. pybind11_add_module(uctc uctc.cc math/arith.cc operators/nn.cc operators/ops.cc tensor/tensor.cc)
  10. add_custom_command(
  11. TARGET uctc POST_BUILD
  12. COMMAND ${CMAKE_COMMAND} -E echo "Changing directory and running Python script for generate interpreter annotations"
  13. COMMAND ${CMAKE_COMMAND} -E chdir ${CMAKE_BINARY_DIR} pybind11-stubgen uctc --output-dir .
  14. )