GpuBackend.h

浏览该文件的文档.

#ifndef EVE_TENSOR_GPUBACKEND_H
#define EVE_TENSOR_GPUBACKEND_H
 
#include <vector>
 
namespace eve::tensor {
 
class Graph;
struct OptimizedGraph;
 
class GpuProgram {
public:
    ~GpuProgram();
 
    static GpuProgram *tryBuild(const Graph &graph, const OptimizedGraph &opt, int outputNode);
 
    std::vector<float> run(const std::vector<const float *> &feeds) const;
 
private:
    GpuProgram();
    GpuProgram(const GpuProgram &) = delete;
    GpuProgram &operator=(const GpuProgram &) = delete;
 
    struct Impl;
    Impl *impl_ = nullptr;
};
 
bool gpuReduce(const float *data, int size, int op, float &outResult);
 
}  // namespace eve::tensor
 
#endif  // EVE_TENSOR_GPUBACKEND_H