Halide 22.0.0 Halide compiler and libraries

InjectHostDevBufferCopies.h
Go to the documentation of this file.
1#ifndef HALIDE_HOST_GPU_BUFFER_COPIES_H
2#define HALIDE_HOST_GPU_BUFFER_COPIES_H
3
4/** \file
5 * Defines the lowering passes that deal with host and device buffer flow.
6 */
7
8#include <string>
9#include <vector>
10
11#include "Expr.h"
12
13namespace Halide {
14
15struct Target;
16
17namespace Internal {
18
19/** A helper function to call an extern function, and assert that it
20 * returns 0. */
21 Stmt call_extern_and_assert(const std::string &name, const std::vector<Expr> &args);
22
23/** Inject calls to halide_device_malloc, halide_copy_to_device, and
24 * halide_copy_to_host as needed. */
25 Stmt inject_host_dev_buffer_copies(Stmt s, const Target &t);
26
27} // namespace Internal
28} // namespace Halide
29
30#endif
Expr.h
Base classes for Halide expressions (Halide::Expr) and statements (Halide::Internal::Stmt)
Halide::Internal::call_extern_and_assert
Stmt call_extern_and_assert(const std::string &name, const std::vector< Expr > &args)
A helper function to call an extern function, and assert that it returns 0.
Halide::Internal::inject_host_dev_buffer_copies
Stmt inject_host_dev_buffer_copies(Stmt s, const Target &t)
Inject calls to halide_device_malloc, halide_copy_to_device, and halide_copy_to_host as needed.
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::Stmt
A reference-counted handle to a statement node.
Definition: Expr.h:427
Halide::Target
A struct representing a target machine and os to generate code for.
Definition: Target.h:19