Halide
SelectGPUAPI.h
Go to the documentation of this file.
1
#ifndef HALIDE_INTERNAL_SELECT_GPU_API_H
2
#define HALIDE_INTERNAL_SELECT_GPU_API_H
3
4
#include "
Expr.h
"
5
6
/** \file
7
* Defines a lowering pass that selects which GPU api to use for each
8
* gpu for loop
9
*/
10
11
namespace
Halide
{
12
13
struct
Target;
14
15
namespace
Internal
{
16
17
/** Replace for loops with GPU_Default device_api with an actual
18
* device API depending on what's enabled in the target. Choose the
19
* first of the following: opencl, cuda, openglcompute, opengl */
20
Stmt
select_gpu_api
(
const
Stmt &s,
const
Target &t);
21
22
}
// namespace Internal
23
}
// namespace Halide
24
25
#endif
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
Halide::Internal::select_gpu_api
Stmt select_gpu_api(const Stmt &s, const Target &t)
Replace for loops with GPU_Default device_api with an actual device API depending on what's enabled i...
src
SelectGPUAPI.h
Generated by
1.8.17