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
src
OffloadGPULoops.h
Generated by
1.8.17