Print TORCH_DIR for debug.

This commit is contained in:
Fangjun Kuang 2022-07-19 18:49:29 +08:00
parent e15b2ae6f5
commit 8384a58c57

View File

@ -8,6 +8,7 @@ execute_process(
OUTPUT_STRIP_TRAILING_WHITESPACE
OUTPUT_VARIABLE TORCH_DIR
)
message(STATUS "TORCH_DIR: ${TORCH_DIR}")
list(APPEND CMAKE_PREFIX_PATH "${TORCH_DIR}")
find_package(Torch REQUIRED)