Provide CMake option to enable Vulkan API.

[ghstack-poisoned]
pytorch · Oct 16, 2020 · 416b39d · 416b39d
1 parent 46db908
commit 416b39d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -202,6 +202,7 @@ cmake_dependent_option(
     USE_VALGRIND "Use Valgrind. Only available on Linux." ON
     "LINUX" OFF)
 option(USE_VULKAN "Use Vulkan GPU backend" OFF)
+option(USE_VULKAN_API "Use Vulkan GPU backend v2" OFF)
 option(USE_VULKAN_WRAPPER "Use Vulkan wrapper" ON)
 option(USE_VULKAN_SHADERC_RUNTIME "Use Vulkan Shader compilation runtime(Needs shaderc lib)" OFF)
 option(USE_VULKAN_RELAXED_PRECISION "Use Vulkan relaxed precision(mediump)" OFF)
@@ -535,6 +536,10 @@ if(USE_VULKAN)
   string(APPEND CMAKE_CXX_FLAGS " -DUSE_VULKAN")
 endif()
 
+if(USE_VULKAN_API)
+  string(APPEND CMAKE_CXX_FLAGS " -DUSE_VULKAN_API")
+endif()
+
 if(USE_VULKAN_WRAPPER)
   string(APPEND CMAKE_CXX_FLAGS " -DUSE_VULKAN_WRAPPER")
 endif()