ITKVkFFTBackend/itk-module-init.cmake at main · InsightSoftwareConsortium/ITKVkFFTBackend · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
#
# Find the packages required by this module
#
if(NOT DEFINED VKFFT_BACKEND)
  include(CheckLanguage)
  check_language(CUDA)
  set(_vkfft_have_metal FALSE)
  if(APPLE)
    find_library(_vkfft_metal_probe Metal)
    find_library(_vkfft_foundation_probe Foundation)
    find_library(_vkfft_quartzcore_probe QuartzCore)
    if(
      _vkfft_metal_probe
      AND
        _vkfft_foundation_probe
      AND
        _vkfft_quartzcore_probe
    )
      set(_vkfft_have_metal TRUE)
    endif()
    unset(_vkfft_metal_probe CACHE)
    unset(_vkfft_foundation_probe CACHE)
    unset(_vkfft_quartzcore_probe CACHE)
  endif()
  if(CMAKE_CUDA_COMPILER)
    set(_vkfft_backend_default 1)
  elseif(_vkfft_have_metal)
    set(_vkfft_backend_default 5)
  else()
    set(_vkfft_backend_default 3)
  endif()
else()
  set(_vkfft_backend_default ${VKFFT_BACKEND})
endif()
set(
  VKFFT_BACKEND
  ${_vkfft_backend_default}
  CACHE STRING
  "0 - Vulkan, 1 - CUDA, 2 - HIP, 3 - OpenCL, 4 - Level Zero, 5 - Metal"
)
if(${VKFFT_BACKEND} EQUAL 1)
  find_package(CUDAToolkit REQUIRED)
  set(CUDA_LIBRARIES CUDA::cudart)
  find_library(
    CUDA_NVRTC_LIB
    libnvrtc
    nvrtc
    HINTS
      "${CUDAToolkit_LIBRARY_DIR}"
      "/usr/lib64"
      "/usr/local/cuda/lib64"
  )
elseif(${VKFFT_BACKEND} EQUAL 3)
  find_package(OpenCL REQUIRED)
elseif(${VKFFT_BACKEND} EQUAL 4)
  find_path(
    LevelZero_INCLUDE_DIR
    NAMES
      level_zero/ze_api.h
    HINTS
    ENV LEVEL_ZERO_ROOT
    ENV CMPLR_ROOT
    PATH_SUFFIXES
      include
  )
  find_library(
    LevelZero_LIBRARY
    NAMES
      ze_loader
    HINTS
    ENV LEVEL_ZERO_ROOT
    ENV CMPLR_ROOT
    PATH_SUFFIXES
      lib
      lib64
      lib/x64
  )
  if(NOT LevelZero_INCLUDE_DIR OR NOT LevelZero_LIBRARY)
    message(
      FATAL_ERROR
      "VKFFT_BACKEND=4 (Level Zero) requires the oneAPI Level Zero loader (ze_loader) and headers (level_zero/ze_api.h)."
    )
  endif()
elseif(${VKFFT_BACKEND} EQUAL 5)
  if(NOT APPLE)
    message(
      FATAL_ERROR
      "VKFFT_BACKEND=5 (Metal) requires Apple platforms (macOS/iOS, including Apple Silicon)."
    )
  endif()
  find_library(METAL_FRAMEWORK Metal REQUIRED)
  find_library(FOUNDATION_FRAMEWORK Foundation REQUIRED)
  find_library(QUARTZCORE_FRAMEWORK QuartzCore REQUIRED)
endif()