Halide  12.0.1
Halide compiler and libraries
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
Base classes for Halide expressions (Halide::Expr) and statements (Halide::Internal::Stmt)
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...
This file defines the class FunctionDAG, which is our representation of a Halide pipeline,...
@ Internal
Not visible externally, similar to 'static' linkage in C.
A reference-counted handle to a statement node.
Definition: Expr.h:413
A struct representing a target machine and os to generate code for.
Definition: Target.h:19