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
|
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
# - Find Arrow CUDA (arrow/gpu/cuda_api.h, libarrow_cuda.a, libarrow_cuda.so)
#
# This module requires Arrow from which it uses
# arrow_find_package()
#
# This module defines
# ARROW_CUDA_FOUND, whether Arrow CUDA has been found
# ARROW_CUDA_IMPORT_LIB, path to libarrow_cuda's import library (Windows only)
# ARROW_CUDA_INCLUDE_DIR, directory containing headers
# ARROW_CUDA_LIBS, deprecated. Use ARROW_CUDA_LIB_DIR instead
# ARROW_CUDA_LIB_DIR, directory containing Arrow CUDA libraries
# ARROW_CUDA_SHARED_IMP_LIB, deprecated. Use ARROW_CUDA_IMPORT_LIB instead
# ARROW_CUDA_SHARED_LIB, path to libarrow_cuda's shared library
# ARROW_CUDA_STATIC_LIB, path to libarrow_cuda.a
if(DEFINED ARROW_CUDA_FOUND)
return()
endif()
set(find_package_arguments)
if(${CMAKE_FIND_PACKAGE_NAME}_FIND_VERSION)
list(APPEND find_package_arguments "${${CMAKE_FIND_PACKAGE_NAME}_FIND_VERSION}")
endif()
if(${CMAKE_FIND_PACKAGE_NAME}_FIND_REQUIRED)
list(APPEND find_package_arguments REQUIRED)
endif()
if(${CMAKE_FIND_PACKAGE_NAME}_FIND_QUIETLY)
list(APPEND find_package_arguments QUIET)
endif()
find_package(Arrow ${find_package_arguments})
if(ARROW_FOUND)
arrow_find_package(ARROW_CUDA
"${ARROW_HOME}"
arrow_cuda
arrow/gpu/cuda_api.h
ArrowCUDA
arrow-cuda)
if(NOT ARROW_CUDA_VERSION)
set(ARROW_CUDA_VERSION "${ARROW_VERSION}")
endif()
endif()
if("${ARROW_CUDA_VERSION}" VERSION_EQUAL "${ARROW_VERSION}")
set(ARROW_CUDA_VERSION_MATCH TRUE)
else()
set(ARROW_CUDA_VERSION_MATCH FALSE)
endif()
mark_as_advanced(ARROW_CUDA_IMPORT_LIB
ARROW_CUDA_INCLUDE_DIR
ARROW_CUDA_LIBS
ARROW_CUDA_LIB_DIR
ARROW_CUDA_SHARED_IMP_LIB
ARROW_CUDA_SHARED_LIB
ARROW_CUDA_STATIC_LIB
ARROW_CUDA_VERSION
ARROW_CUDA_VERSION_MATCH)
find_package_handle_standard_args(
ArrowCUDA
REQUIRED_VARS ARROW_CUDA_INCLUDE_DIR ARROW_CUDA_LIB_DIR ARROW_CUDA_VERSION_MATCH
VERSION_VAR ARROW_CUDA_VERSION)
set(ARROW_CUDA_FOUND ${ArrowCUDA_FOUND})
if(ArrowCUDA_FOUND AND NOT ArrowCUDA_FIND_QUIETLY)
message(STATUS "Found the Arrow CUDA by ${ARROW_CUDA_FIND_APPROACH}")
message(STATUS "Found the Arrow CUDA shared library: ${ARROW_CUDA_SHARED_LIB}")
message(STATUS "Found the Arrow CUDA import library: ${ARROW_CUDA_IMPORT_LIB}")
message(STATUS "Found the Arrow CUDA static library: ${ARROW_CUDA_STATIC_LIB}")
endif()
|
