MNN/codegen/opencl/OpenCLTarget.hpp

//
//  OpenCLTarget.hpp
//  MNN
//
//  Created by MNN on 2022/11/14.
//  Copyright © 2018, Alibaba Group Holding Limited
//
#include "../SourceModule.hpp"
namespace MNN {

class OpenCLTarget : public Target {
public:
    OpenCLTarget(const BackendConfig::PrecisionMode precision) : Target(precision) {}
    ~OpenCLTarget() {}
    std::string codegen(std::vector<std::string>& inputs, const Command* cmd, std::string& inpName) override;
private:
    std::string type() override;
    std::string macro() override;
    std::string number(float val) override;
    std::string load(const std::string& base, const std::string& offset, const Command* cmd, std::string& inpName) override;
    std::string loadscalar(const std::string& base, std::string& inpName) override;
    std::string store(const std::string base, const std::string& offset, const std::string& data) override;
    std::string proto(const std::string& name, const std::vector<std::string>& inputs, const std::vector<std::string>& outputs, bool hasSingleConvertRaster = false) override;
    template <typename T>
    std::string numval(T t) { return "((FLOAT4)" + std::to_string(t) + ")"; }
};

}