Skip to content

[Vulkan] Add device capabilities to Target, use in codegen #8127

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
masahi merged 9 commits into from
May 26, 2021
Merged

[Vulkan] Add device capabilities to Target, use in codegen #8127

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
56f28ac
[Vulkan] Enable instance/device extensions
Lunderberg May 21, 2021
1882c27
[Vulkan] Extract device information from vulkan API.
Lunderberg May 24, 2021
ef731c2
[Vulkan] Query instance-supported apiVersion before creating instance
Lunderberg May 24, 2021
fdf16c3
[Vulkan] Moved options for dedicated allocation and push descriptors …
Lunderberg May 24, 2021
cada3fd
[Vulkan] Move option for vulkan validation layers to environment vari…
Lunderberg May 24, 2021
457119b
[Vulkan] Explicitly enable vulkan features in device creation
Lunderberg May 24, 2021
32741fe
[Vulkan] Avoid repeated queries for device attributes.
Lunderberg May 24, 2021
d95e77e
[Vulkan] Implement "from_device" flag for the vulkan target.
Lunderberg May 25, 2021
1d262d7
[Vulkan][Codegen] Read vulkan device capabilities/limits from Target
Lunderberg May 25, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
21 changes: 0 additions & 21 deletions cmake/modules/Vulkan.cmake
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,14 +18,6 @@
# Be compatible with older version of CMake
find_vulkan(${USE_VULKAN})

# Extra Vulkan runtime options, exposed for advanced users.
tvm_option(USE_VULKAN_IMMEDIATE_MODE "Use Vulkan Immediate mode
(KHR_push_descriptor extension)" ON IF USE_VULKAN)
tvm_option(USE_VULKAN_DEDICATED_ALLOCATION "Use Vulkan dedicated allocations" ON
IF USE_VULKAN)
tvm_option(USE_VULKAN_VALIDATION "Enable Vulkan API validation layers" OFF
IF USE_VULKAN)

if(USE_VULKAN)
if(NOT Vulkan_FOUND)
message(FATAL_ERROR "Cannot find Vulkan, USE_VULKAN=" ${USE_VULKAN})
Expand All @@ -38,17 +30,4 @@ if(USE_VULKAN)
list(APPEND COMPILER_SRCS ${COMPILER_VULKAN_SRCS})
list(APPEND TVM_LINKER_LIBS ${Vulkan_SPIRV_TOOLS_LIBRARY})
list(APPEND TVM_RUNTIME_LINKER_LIBS ${Vulkan_LIBRARY})

if(USE_VULKAN_IMMEDIATE_MODE)
message(STATUS "Build with Vulkan immediate mode")
add_definitions(-DUSE_VULKAN_IMMEDIATE_MODE=1)
endif()
if(USE_VULKAN_DEDICATED_ALLOCATION)
message(STATUS "Build with Vulkan dedicated allocation")
add_definitions(-DUSE_VULKAN_DEDICATED_ALLOCATION=1)
endif()
if(USE_VULKAN_VALIDATION)
message(STATUS "Build with Vulkan API validation")
add_definitions(-DUSE_VULKAN_VALIDATION=1)
endif()
endif(USE_VULKAN)
2 changes: 1 addition & 1 deletion python/tvm/_ffi/runtime_ctypes.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -376,7 +376,7 @@ def api_version(self):
The version of the SDK

"""
return self._GetDeviceAttr(self.device_type, self.device_id, 12)
return self._GetDeviceAttr(self.device_type, self.device_id, 11)

@property
def driver_version(self):
Expand Down
Loading