Add Queue::submit() for command buffer submission

include/API/CommandBuffer.h

@@ -0,0 +1,44 @@
+//===- CommandBuffer.h - Offload Command Buffer API -----------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+//
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef OFFLOADTEST_API_COMMANDBUFFER_H
+#define OFFLOADTEST_API_COMMANDBUFFER_H
+
+#include "API/API.h"
+
+#include <cassert>
+
+namespace offloadtest {
+
+class CommandBuffer {
+  GPUAPI API;
+
+public:
+  explicit CommandBuffer(GPUAPI API) : API(API) {}
+  virtual ~CommandBuffer() = default;
+  CommandBuffer(const CommandBuffer &) = delete;
+  CommandBuffer &operator=(const CommandBuffer &) = delete;
+
+  GPUAPI getAPI() const { return API; }
+
+  template <typename T> T &as() {
+    assert(API == T::BackendAPI && "CommandBuffer backend mismatch");
+    return static_cast<T &>(*this);
+  }
+  template <typename T> const T &as() const {
+    assert(API == T::BackendAPI && "CommandBuffer backend mismatch");
+    return static_cast<const T &>(*this);
+  }
+};
+
+} // namespace offloadtest
+
+#endif // OFFLOADTEST_API_COMMANDBUFFER_H

include/API/Device.h

@@ -16,8 +16,10 @@
 
 #include "API/API.h"
 #include "API/Capabilities.h"
+#include "API/CommandBuffer.h"
 #include "llvm/ADT/StringRef.h"
 #include "llvm/ADT/iterator_range.h"
+#include "llvm/Support/Error.h"
 
 #include <memory>
 #include <string>
@@ -60,6 +62,19 @@ class Queue {
 public:
   virtual ~Queue() = 0;
 
+  /// Submit command buffers for execution and block until completion.
+  // TODO: Return a Fence instead of blocking, once the Fence abstraction
+  // from PR #1007 is available.
+  virtual llvm::Error
+  submit(llvm::SmallVector<std::unique_ptr<CommandBuffer>> CBs) = 0;
+
+  /// Convenience overload for submitting a single command buffer.
+  llvm::Error submit(std::unique_ptr<CommandBuffer> CB) {
+    llvm::SmallVector<std::unique_ptr<CommandBuffer>> CBs;
+    CBs.push_back(std::move(CB));
+    return submit(std::move(CBs));
+  }
+
 protected:
   Queue() = default;
 };
@@ -82,6 +97,12 @@ class Device {
                size_t SizeInBytes) = 0;
   virtual void printExtra(llvm::raw_ostream &OS) {}
 
+  virtual llvm::Expected<std::unique_ptr<CommandBuffer>> createCommandBuffer() {
+    return llvm::createStringError(
+        std::errc::not_supported,
+        "createCommandBuffer not implemented for this backend");
+  }
+
   virtual ~Device() = 0;
 
   llvm::StringRef getDescription() const { return Description; }