dispatch_diagnostics.hpp

Go to the documentation of this file.

#pragma once
 
#include <uni20/linalg/dispatch_error.hpp>
 
#include <atomic>
#include <functional>
#include <optional>
#include <string>
#include <string_view>
#include <utility>
#include <vector>
 
namespace uni20::linalg::dispatch_diagnostics
{
 
struct event
{
    std::string operation;
    std::vector<KernelBackendAttempt> backend_attempts;
 
    [[nodiscard]] bool succeeded() const noexcept
    {
      for (auto const& attempt : backend_attempts)
      {
        if (attempt.runtime_result && kernel_attempt_succeeded(*attempt.runtime_result)) return true;
      }
      return false;
    }
 
    [[nodiscard]] std::optional<std::string_view> selected_backend() const noexcept
    {
      for (auto const& attempt : backend_attempts)
      {
        if (attempt.runtime_result && kernel_attempt_succeeded(*attempt.runtime_result)) return attempt.backend;
      }
      return std::nullopt;
    }
};
 
using sink = std::function<void(event const&)>;
 
namespace detail
{
extern std::atomic<bool> enabled_flag;
}
 
[[nodiscard]] inline bool enabled() noexcept { return detail::enabled_flag.load(std::memory_order_relaxed); }
 
void set_sink(sink replacement);
 
void reset_sink();
 
[[nodiscard]] sink current_sink();
 
class scoped_sink {
  public:
    explicit scoped_sink(sink replacement);
    scoped_sink(scoped_sink const&) = delete;
    scoped_sink& operator=(scoped_sink const&) = delete;
    scoped_sink(scoped_sink&& other) noexcept;
    scoped_sink& operator=(scoped_sink&& other) noexcept;
    ~scoped_sink();
 
  private:
    sink previous_;
    bool active_ = true;
};
 
namespace detail
{
void emit(event const& diagnostic);
} // namespace detail
 
} // namespace uni20::linalg::dispatch_diagnostics