linalg/precision.hpp

Go to the documentation of this file.

#pragma once
 
#include <uni20/core/runtime_precision.hpp>
#include <uni20/linalg/dispatch.hpp>
 
namespace uni20::linalg::detail
{
template <class S, class Selector, class Operation, class... Args>
  requires((has_runtime_precision_v<S> &&
            requires(Selector&& selector, Operation const& op, Precision p, Args&&... args) {
              dispatch_kernel(std::forward<Selector>(selector), op, std::forward<Args>(args)..., p);
            }) ||
           (!has_runtime_precision_v<S> &&
            requires(Selector&& selector, Operation const& op, Args&&... args) {
              dispatch_kernel(std::forward<Selector>(selector), op, std::forward<Args>(args)...);
            }))
void dispatch_at_precision(Selector&& selector, Operation const& operation, Precision p, Args&&... args)
{
  require_precision<S>(p);
  if constexpr (has_runtime_precision_v<S>)
    dispatch_kernel(std::forward<Selector>(selector), operation, std::forward<Args>(args)..., p);
  else
    dispatch_kernel(std::forward<Selector>(selector), operation, std::forward<Args>(args)...);
}
} // namespace uni20::linalg::detail