https://github.com/halide/Halide
Raw File
Tip revision: 72c91b18a594318c2323bc9668d05e9f85748cb1 authored by Andrew Adams on 14 January 2020, 01:09:30 UTC
Switch to 16 entries per line
Tip revision: 72c91b1
user_context_generator.cpp
#include "Halide.h"

using namespace Halide;

class UserContextGenerator : public Halide::Generator<UserContextGenerator> {
public:
    Input<uint8_t> constant{"constant"};
    Output<Buffer<uint8_t>> output{"output", 1};

    Var x;

    void generate() {
        output(x) = constant;
    }

    void schedule() {
    }
};

HALIDE_REGISTER_GENERATOR(UserContextGenerator, user_context)
back to top