clover: Add initial implementation of clCompileProgram for CL 1.2.

[mesa.git] / src / gallium / state_trackers / clover / core / program.cpp

diff --git a/src/gallium/state_trackers/clover/core/program.cpp b/src/gallium/state_trackers/clover/core/program.cpp
index 8082cf0f6f6942e554bb66a913e71a1b49ef3bab..8553ca7ec96c37104597ce2c7372e1787e89f100 100644 (file)
--- a/src/gallium/state_trackers/clover/core/program.cpp
+++ b/src/gallium/state_trackers/clover/core/program.cpp
@@ -21,18 +21,18 @@
 //
 
 #include "core/program.hpp"
-#include "core/compiler.hpp"
 
 using namespace clover;
 
-program::program(context &ctx, const std::string &source) :
-   ctx(ctx), _source(source) {
+program::program(clover::context &ctx, const std::string &source) :
+   has_source(true), context(ctx), _source(source), _kernel_ref_counter(0) {
 }
 
-program::program(context &ctx,
+program::program(clover::context &ctx,
                  const ref_vector<device> &devs,
                  const std::vector<module> &binaries) :
-   ctx(ctx) {
+   has_source(false), context(ctx),
+   _devices(devs), _kernel_ref_counter(0) {
    for_each([&](device &dev, const module &bin) {
          _binaries.insert({ &dev, bin });
       },
@@ -40,24 +40,33 @@ program::program(context &ctx,
 }
 
 void
-program::build(const ref_vector<device> &devs, const char *opts) {
-   for (auto &dev : devs) {
-      _binaries.erase(&dev);
-      _logs.erase(&dev);
-      _opts.erase(&dev);
-
-      _opts.insert({ &dev, opts });
-
-      try {
-         auto module = (dev.ir_format() == PIPE_SHADER_IR_TGSI ?
-                        compile_program_tgsi(_source) :
-                        compile_program_llvm(_source, dev.ir_format(),
-                                             dev.ir_target(), build_opts(dev)));
-         _binaries.insert({ &dev, module });
-
-      } catch (build_error &e) {
-         _logs.insert({ &dev, e.what() });
-         throw;
+program::build(const ref_vector<device> &devs, const char *opts,
+               const header_map &headers) {
+   if (has_source) {
+      _devices = devs;
+
+      for (auto &dev : devs) {
+         _binaries.erase(&dev);
+         _logs.erase(&dev);
+         _opts.erase(&dev);
+
+         _opts.insert({ &dev, opts });
+
+         compat::string log;
+
+         try {
+            auto module = (dev.ir_format() == PIPE_SHADER_IR_TGSI ?
+                           compile_program_tgsi(_source) :
+                           compile_program_llvm(_source, headers,
+                                                dev.ir_format(),
+                                                dev.ir_target(), build_opts(dev),
+                                                log));
+            _binaries.insert({ &dev, module });
+            _logs.insert({ &dev, log });
+         } catch (const build_error &) {
+            _logs.insert({ &dev, log });
+            throw;
+         }
       }
    }
 }
@@ -67,22 +76,43 @@ program::source() const {
    return _source;
 }
 
-const std::map<device *, module> &
-program::binaries() const {
-   return _binaries;
+program::device_range
+program::devices() const {
+   return map(evals(), _devices);
+}
+
+const module &
+program::binary(const device &dev) const {
+   return _binaries.find(&dev)->second;
 }
 
 cl_build_status
-program::build_status(device &dev) const {
-   return _binaries.count(&dev) ? CL_BUILD_SUCCESS : CL_BUILD_NONE;
+program::build_status(const device &dev) const {
+   if (_binaries.count(&dev))
+      return CL_BUILD_SUCCESS;
+   else
+      return CL_BUILD_NONE;
 }
 
 std::string
-program::build_opts(device &dev) const {
+program::build_opts(const device &dev) const {
    return _opts.count(&dev) ? _opts.find(&dev)->second : "";
 }
 
 std::string
-program::build_log(device &dev) const {
+program::build_log(const device &dev) const {
    return _logs.count(&dev) ? _logs.find(&dev)->second : "";
 }
+
+const compat::vector<module::symbol> &
+program::symbols() const {
+   if (_binaries.empty())
+      throw error(CL_INVALID_PROGRAM_EXECUTABLE);
+
+   return _binaries.begin()->second.syms;
+}
+
+unsigned
+program::kernel_ref_count() const {
+   return _kernel_ref_counter.ref_count();
+}