OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Error if pocketfft is not found (#67909)

Browse Source 
Summary:
Fixes https://github.com/pytorch/pytorch/issues/67842

cc mruberry peterbell10

Pull Request resolved: https://github.com/pytorch/pytorch/pull/67909

Reviewed By: albanD

Differential Revision: D33759534

Pulled By: malfet

fbshipit-source-id: 03548c95fe233b812b303ce9603c20ff9f626c39
This commit is contained in:
Peter Bell 2022-01-31 09:23:55 -08:00 committed by Facebook GitHub Bot
parent 3a53b3e94f
commit 214624e254
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
cmake/Dependencies.cmake
View File

@ -293,15 +293,15 @@ endif()
# --- [ PocketFFT
set(AT_POCKETFFT_ENABLED 0)
if(NOT AT_MKL_ENABLED)
find_path(POCKETFFT_INCLUDE_DIR NAMES pocketfft_hdronly.h PATHS
/usr/local/include
ENV POCKETFFT_HOME
"${PROJECT_SOURCE_DIR}/third_party/pocketfft"
)
if(POCKETFFT_INCLUDE_DIR)
set(AT_POCKETFFT_ENABLED 1)
message(STATUS "Using pocketfft in directory: ${POCKETFFT_INCLUDE_DIR}")
set(POCKETFFT_INCLUDE_DIR "${Torch_SOURCE_DIR}/third_party/pocketfft/")
if(NOT EXISTS "${POCKETFFT_INCLUDE_DIR}")
message(FATAL_ERROR "pocketfft directory not found, expected ${POCKETFFT_INCLUDE_DIR}")
elif(NOT EXISTS "${POCKETFFT_INCLUDE_DIR}/pocketfft_hdronly.h")
message(FATAL_ERROR "pocketfft headers not found in ${POCKETFFT_INCLUDE_DIR}")
endif()
set(AT_POCKETFFT_ENABLED 1)
message(STATUS "Using pocketfft in directory: ${POCKETFFT_INCLUDE_DIR}")
endif()
# ---[ Dependencies