Vulkan2 Runtime API

Author: ajtulloch

cmake/modules/Vulkan.cmake

@@ -29,7 +29,7 @@ if(USE_VULKAN)
     message(FATAL_ERROR "Cannot find Vulkan, USE_VULKAN=" ${USE_VULKAN})
   endif()
   message(STATUS "Build with VULKAN support")
-  file(GLOB RUNTIME_VULKAN_SRCS src/runtime/vulkan/*.cc)
+  file(GLOB RUNTIME_VULKAN_SRCS src/runtime/vulkan/vulkan2.cc)
   file(GLOB COMPILER_VULKAN_SRCS src/codegen/spirv/*.cc)
   list(APPEND RUNTIME_SRCS ${RUNTIME_VULKAN_SRCS})
   list(APPEND COMPILER_SRCS ${COMPILER_VULKAN_SRCS})

src/codegen/spirv/build_vulkan.cc

@@ -29,7 +29,9 @@
 
 #include "codegen_spirv.h"
 #include "../build_common.h"
-#include "../../runtime/vulkan/vulkan_module.h"
+
+#include "../../runtime/vulkan/vulkan2_shader.h"
+#include "../../runtime/vulkan/vulkan2_module.h"
 
 namespace tvm {
 namespace codegen {

src/codegen/spirv/ir_builder.cc

@@ -33,7 +33,10 @@ namespace spirv {
 void IRBuilder::InitHeader() {
   CHECK_EQ(header_.size(), 0U);
   header_.push_back(spv::MagicNumber);
-  header_.push_back(spv::Version);
+  // Use SPIR-V v1.0. This needs to be kept in sync (or at least behind)
+  // `VkApplicationInfo.apiVersion` in `vulkan2.cc` to ensure Vulkan API
+  // validation passes.
+  header_.push_back(0x10000);
   // generator: set to 0, unknown
   header_.push_back(0U);
   // Bound: set during Finalize

src/runtime/vulkan/README.md

@@ -0,0 +1,36 @@
+
+Key components
+
+
+VulkanStream abstraction. Stream abstraction supports several operations:
+
+- stream->launch(function, args, ..)
+
+This maps to:
+
+- obtain current command buffer
+- dispatch
+- barrier -> (compute|transfer)
+
+
+- stream->synchronize();
+
+This maps to
+
+- vkEndCommandBuffer
+- vkQueueSubmit
+- vkQueueWaitIdle
+
+
+
+- stream->copy(from=VULKAN, to=VULKAN)
+- stream->copy(from=CPU, to=VULKAN)
+- stream->copy(from=VULKAN, to=CPU)
+
+This abstracts over CommandBuffer, CommandBufferPool, Fence, etc.
+
+
+Then, API command are fairly simple, and mirror existing CUDA design.
+
+The key approach is:
+