contraction_precision.h

Go to the documentation of this file.

#pragma once
 
#include <string>
 
#include "sdfg/analysis/analysis.h"
#include "sdfg/builder/structured_sdfg_builder.h"
#include "sdfg/fp_math/transformations/ozaki_contraction.h"
#include "sdfg/passes/pass.h"
 
namespace sdfg {
namespace fp_math {

struct ContractionPrecision {
    enum class Kind { Native, Quantize, Ozaki };
 
    Kind kind = Kind::Native;
    OzakiVariant variant = OzakiVariant::Modular;
    int parameter = 0;
    Rescaling rescaling = Rescaling::Exponents;

    static ContractionPrecision parse(const std::string& spec);
 
    std::string str() const;
};

class ContractionPrecisionPass : public passes::Pass {
private:
    ContractionPrecision precision_;
 
public:
    explicit ContractionPrecisionPass(const ContractionPrecision& precision);
 
    virtual std::string name() override;
 
    virtual bool run_pass(builder::StructuredSDFGBuilder& builder, analysis::AnalysisManager& analysis_manager) override;
};
 
} // namespace fp_math
} // namespace sdfg