mirror of https://github.com/alibaba/MNN.git
29 lines
1.1 KiB
C++
29 lines
1.1 KiB
C++
//
|
|
// 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) + ")"; }
|
|
};
|
|
|
|
}
|