Halide
PythonExtensionGen.h
Go to the documentation of this file.
1 #ifndef HALIDE_PYTHON_EXTENSION_GEN_H_
2 #define HALIDE_PYTHON_EXTENSION_GEN_H_
3 
4 #include "Module.h"
5 #include "Target.h"
6 #include <string>
7 
8 namespace Halide {
9 namespace Internal {
10 
12 public:
13  PythonExtensionGen(std::ostream &dest);
14 
15  void compile(const Module &module);
16 
17 private:
18  std::ostream &dest;
19  std::vector<std::string> buffer_refs;
20 
21  void compile(const LoweredFunc &f);
22  void convert_buffer(const std::string &name, const LoweredArgument *arg);
23  void release_buffers(const std::string &prefix);
24 };
25 
26 } // namespace Internal
27 } // namespace Halide
28 
29 #endif // HALIDE_PYTHON_EXTENSION_GEN_H_
Target.h
Halide::Module
A halide module.
Definition: Module.h:136
Halide::Internal::PythonExtensionGen::compile
void compile(const Module &module)
Halide
This file defines the class FunctionDAG, which is our representation of a Halide pipeline,...
Definition: AddAtomicMutex.h:21
Halide::LinkageType::Internal
@ Internal
Not visible externally, similar to 'static' linkage in C.
Halide::Internal::PythonExtensionGen
Definition: PythonExtensionGen.h:11
Halide::Internal::LoweredFunc
Definition of a lowered function.
Definition: Module.h:97
Halide::Internal::PythonExtensionGen::PythonExtensionGen
PythonExtensionGen(std::ostream &dest)
Module.h
Halide::Internal::LoweredArgument
Definition of an argument to a LoweredFunc.
Definition: Module.h:80