convenience.h

Go to the documentation of this file.

#pragma once
 
#include "pipeline/compressor.h"
#include "predictors/lorenzo/lorenzo.h"
 
namespace fz {
 
template <typename TInput = float, typename TCode = uint16_t>
inline LorenzoStage<TInput, TCode>* addLorenzo(
    Pipeline& pipeline,
    float     error_bound,
    int       quant_radius     = 32768,
    float     outlier_capacity = 0.2f
) {
    typename LorenzoStage<TInput, TCode>::Config cfg;
    cfg.error_bound      = error_bound;
    cfg.quant_radius     = quant_radius;
    cfg.outlier_capacity = outlier_capacity;
    cfg.dims             = pipeline.getDims();
    return pipeline.addStage<LorenzoStage<TInput, TCode>>(cfg);
}
 
} // namespace fz