Halide
OffloadGPULoops.h
Go to the documentation of this file.
1 #ifndef HALIDE_OFFLOAD_GPU_LOOPS_H
2 #define HALIDE_OFFLOAD_GPU_LOOPS_H
3 
4 /** \file
5  * Defines a lowering pass to pull loops marked with
6  * GPU device APIs to a separate module, and call them through the
7  * appropriate host runtime module.
8  */
9 
10 #include "Expr.h"
11 
12 namespace Halide {
13 
14 struct Target;
15 
16 namespace Internal {
17 
18 /** Pull loops marked with GPU device APIs to a separate
19  * module, and call them through the appropriate host runtime module. */
20 Stmt inject_gpu_offload(const Stmt &s, const Target &host_target);
21 
22 } // namespace Internal
23 } // namespace Halide
24 
25 #endif
Halide::Internal::inject_gpu_offload
Stmt inject_gpu_offload(const Stmt &s, const Target &host_target)
Pull loops marked with GPU device APIs to a separate module, and call them through the appropriate ho...
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.
Expr.h