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 <iostream>
5 #include <string>
6 #include <vector>
7 
8 namespace Halide {
9 
10 class Module;
11 
12 namespace Internal {
13 
14 struct LoweredArgument;
15 struct LoweredFunc;
16 
18 public:
19  PythonExtensionGen(std::ostream &dest);
20 
21  void compile(const Module &module);
22 
23 private:
24  std::ostream &dest;
25 
26  void compile(const LoweredFunc &f);
27 };
28 
29 } // namespace Internal
30 } // namespace Halide
31 
32 #endif // HALIDE_PYTHON_EXTENSION_GEN_H_
Halide::Module
A halide module.
Definition: Module.h:138
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: AbstractGenerator.h:19
Halide::LinkageType::Internal
@ Internal
Not visible externally, similar to 'static' linkage in C.
Halide::Internal::PythonExtensionGen
Definition: PythonExtensionGen.h:17
Halide::Internal::LoweredFunc
Definition of a lowered function.
Definition: Module.h:97
Halide::Internal::PythonExtensionGen::PythonExtensionGen
PythonExtensionGen(std::ostream &dest)