1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
import halide as hl
x = hl.Var("x")
y = hl.Var("y")
@hl.generator(name="simplepy")
class SimplePy:
offset = hl.GeneratorParam(0)
buffer_input = hl.InputBuffer(hl.UInt(8), 2)
# We can use `float` as an alias for `hl.Float(32)` if we like.
float_arg = hl.InputScalar(float)
simple_output = hl.OutputBuffer(hl.Float(32), 2)
def generate(self):
g = self
g.simple_output[x, y] = hl.f32(g.offset + g.buffer_input[x, y]) + g.float_arg
g.simple_output.compute_root()
if __name__ == "__main__":
hl.main()
|
