Skip to content

Commit

Permalink
add option to control AUTOGEN_CHECK_CODE env
Browse files Browse the repository at this point in the history
  • Loading branch information
CoolKbh committed Aug 2, 2024
1 parent 89dc169 commit 35be57f
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions dipu/torch_dipu/csrc_dipu/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -50,6 +50,11 @@ set(AUTOGEN_DIOPI_WRAPPER_DIR "${PROJECT_SOURCE_DIR}/scripts/autogen_diopi_wrapp
set(AUTOGEN_CODE_SH "${AUTOGEN_DIOPI_WRAPPER_DIR}/autogen_wrapped_code.sh")
set(AUTOGEN_CODE_REMOVE_CHECK False)

option(AUTOGEN_CHECK_CODE "Whether to remove check code from autogenkernel" OFF)
if(AUTOGEN_CODE_REMOVE_CHECK)
set(AUTOGEN_CODE_REMOVE_CHECK True)
endif()

# The following parameters are used to auto_generated code.
set(GENERATED_KERNELS "${CMAKE_CURRENT_SOURCE_DIR}/aten/ops/AutoGenedKernels.cpp")
set(GENERATED_KERNELS_VENDOR "${PROJECT_SOURCE_DIR}/third_party/DIOPI/impl/${UsedVendor}/convert_config.yaml")
Expand Down

0 comments on commit 35be57f

Please sign in to comment.