Merge pull request #238 from CNugteren/gemm_api_with_temp_buffer

GEMM API with optional temp buffer

1 files changed, 20 insertions, 0 deletions

diff --git a/src/routine.hpp b/src/routine.hpp
index a8f1cb6a..00f7b5cc 100644
--- a/src/routine.hpp
+++ b/src/routine.hpp
@@ -33,6 +33,26 @@ namespace clblast {
 class Routine {
  public:
 
+  static void InitDatabase(const Device &device, const std::vector<std::string> &kernel_names,
+                           const Precision precision, const std::vector<database::DatabaseEntry> &userDatabase,
+                           Databases &db) {
+    const auto platform_id = device.PlatformID();
+    for (const auto &kernel_name : kernel_names) {
+
+      // Queries the cache to see whether or not the kernel parameter database is already there
+      bool has_db;
+      db(kernel_name) = DatabaseCache::Instance().Get(DatabaseKeyRef{platform_id, device(), precision, kernel_name},
+                                                       &has_db);
+      if (has_db) { continue; }
+
+      // Builds the parameter database for this device and routine set and stores it in the cache
+      log_debug("Searching database for kernel '" + kernel_name + "'");
+      db(kernel_name) = Database(device, kernel_name, precision, userDatabase);
+      DatabaseCache::Instance().Store(DatabaseKey{platform_id, device(), precision, kernel_name},
+                                      Database{db(kernel_name)});
+    }
+  }
+
   // Base class constructor. The user database is an optional extra database to override the
   // built-in database.
   // All heavy preparation work is done inside this constructor.