https://github.com/halide/Halide
Revision cd95b27d6b969a1da8337ba820ad8d0c50884bff authored by Steven Johnson on 27 July 2018, 16:56:13 UTC, committed by GitHub on 27 July 2018, 16:56:13 UTC
Rework runtime to allow more than 64 Target::Features (Issue #2911)
2 parent s d91647c + d734dc0
Raw File
Tip revision: cd95b27d6b969a1da8337ba820ad8d0c50884bff authored by Steven Johnson on 27 July 2018, 16:56:13 UTC
Merge pull request #3116 from halide/srj-features
Tip revision: cd95b27
InjectHostDevBufferCopies.h
#ifndef HALIDE_HOST_GPU_BUFFER_COPIES_H
#define HALIDE_HOST_GPU_BUFFER_COPIES_H

/** \file
 * Defines the lowering passes that deal with host and device buffer flow.
 */

#include "IR.h"
#include "Target.h"

namespace Halide {
namespace Internal {

/** A helper function to call an extern function, and assert that it
 * returns 0. */
Stmt call_extern_and_assert(const std::string &name, const std::vector<Expr> &args);

/** Inject calls to halide_device_malloc, halide_copy_to_device, and
 * halide_copy_to_host as needed. */
Stmt inject_host_dev_buffer_copies(Stmt s, const Target &t);

}  // namespace Internal
}  // namespace Halide

#endif
back to top