Skip to content

Commit

Permalink
pnnx build silent warnings from torch header (#5828)
Browse files Browse the repository at this point in the history
  • Loading branch information
nihui authored Dec 18, 2024
1 parent 4b0d2de commit ad6d84e
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion tools/pnnx/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,7 @@ option(PNNX_COVERAGE "build for coverage" OFF)

# set(Torch_INSTALL_DIR "/home/nihui/osd/pnnx/install" CACHE STRING "")
# set(TorchVision_INSTALL_DIR "/home/nihui/osd/pnnx/install" CACHE STRING "")
# set(protobuf_DIR "/home/nihui/osd/pnnx/install/lib64/cmake/protobuf" CACHE STRING "")

# set(Torch_DIR "${Torch_INSTALL_DIR}/share/cmake/Torch")
set(TorchVision_DIR "${TorchVision_INSTALL_DIR}/share/cmake/TorchVision")
Expand Down Expand Up @@ -77,7 +78,7 @@ else()
message(WARNING "Building without TorchVision")
endif()

include_directories(${TORCH_INCLUDE_DIRS})
include_directories(SYSTEM ${TORCH_INCLUDE_DIRS})

if("${CMAKE_CXX_COMPILER_ID}" STREQUAL "GNU")
# test if libtorch and protobuf has the same cxxabi version
Expand Down

0 comments on commit ad6d84e

Please sign in to comment.