wehub-resource-sync
60 lines
2.5 KiB
CMake
60 lines
2.5 KiB
CMake
# 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.
|
|
|
|
# Be compatible with older version of CMake
|
|
find_vulkan(${USE_VULKAN} ${USE_KHRONOS_SPIRV})
|
|
|
|
if(USE_VULKAN)
|
|
if(NOT Vulkan_FOUND)
|
|
message(FATAL_ERROR "Cannot find Vulkan, USE_VULKAN=" ${USE_VULKAN})
|
|
endif()
|
|
if(USE_SPIRV_KHR_INTEGER_DOT_PRODUCT)
|
|
add_definitions(-DTVM_SPIRV_KHR_INTEGER_DOT_PRODUCT=1)
|
|
message(STATUS "Enable SPIRV_KHR_INTEGER_DOT_PRODUCT")
|
|
endif()
|
|
include_directories(SYSTEM ${Vulkan_INCLUDE_DIRS})
|
|
message(STATUS "Build with Vulkan support")
|
|
tvm_file_glob(GLOB COMPILER_VULKAN_SRCS
|
|
src/backend/vulkan/codegen/build_vulkan.cc
|
|
src/backend/vulkan/codegen/codegen_spirv.cc
|
|
src/backend/vulkan/codegen/intrin_rule_spirv.cc
|
|
src/backend/vulkan/codegen/ir_builder.cc
|
|
src/backend/vulkan/codegen/spirv_support.cc
|
|
src/backend/vulkan/codegen/spirv_utils.cc
|
|
)
|
|
list(APPEND COMPILER_SRCS ${COMPILER_VULKAN_SRCS})
|
|
list(APPEND TVM_LINKER_LIBS ${Vulkan_SPIRV_TOOLS_LIBRARY})
|
|
add_definitions(-DTVM_ENABLE_SPIRV=1)
|
|
endif(USE_VULKAN)
|
|
|
|
if(USE_VULKAN)
|
|
message(STATUS "Build vulkan device runtime")
|
|
|
|
tvm_file_glob(GLOB RUNTIME_VULKAN_SRCS src/backend/vulkan/runtime/*.cc)
|
|
|
|
add_library(tvm_runtime_vulkan_objs OBJECT ${RUNTIME_VULKAN_SRCS})
|
|
target_link_libraries(tvm_runtime_vulkan_objs PUBLIC tvm_ffi_header)
|
|
set_target_properties(tvm_runtime_vulkan_objs PROPERTIES POSITION_INDEPENDENT_CODE ON)
|
|
if(TVM_VISIBILITY_FLAG)
|
|
target_compile_options(tvm_runtime_vulkan_objs PRIVATE "${TVM_VISIBILITY_FLAG}")
|
|
endif()
|
|
add_library(tvm_runtime_vulkan SHARED $<TARGET_OBJECTS:tvm_runtime_vulkan_objs>)
|
|
list(APPEND TVM_RUNTIME_BACKEND_LIBS tvm_runtime_vulkan)
|
|
target_link_libraries(tvm_runtime_vulkan PUBLIC tvm_runtime ${Vulkan_LIBRARY})
|
|
tvm_configure_target_library(tvm_runtime_vulkan RUNTIME_MODULE)
|
|
endif(USE_VULKAN)
|