Add forwarding for the recently-added Func::output_type() method (hal…

…ide#6741)
ardier · Mar 3, 2024 · 6158e04 · 6158e04
1 parent b656c68
commit 6158e04
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/Generator.h b/src/Generator.h
@@ -1881,6 +1881,7 @@ class GeneratorInput_Func : public GeneratorInputImpl<T, Func> {
     HALIDE_FORWARD_METHOD_CONST(Func, defined)
     HALIDE_FORWARD_METHOD_CONST(Func, has_update_definition)
     HALIDE_FORWARD_METHOD_CONST(Func, num_update_definitions)
+    HALIDE_FORWARD_METHOD_CONST(Func, output_type)
     HALIDE_FORWARD_METHOD_CONST(Func, output_types)
     HALIDE_FORWARD_METHOD_CONST(Func, outputs)
     HALIDE_FORWARD_METHOD_CONST(Func, rvars)
@@ -2235,6 +2236,7 @@ class GeneratorOutputBase : public GIOBase {
     HALIDE_FORWARD_METHOD(Func, in)
     HALIDE_FORWARD_METHOD(Func, memoize)
     HALIDE_FORWARD_METHOD_CONST(Func, num_update_definitions)
+    HALIDE_FORWARD_METHOD_CONST(Func, output_type)
     HALIDE_FORWARD_METHOD_CONST(Func, output_types)
     HALIDE_FORWARD_METHOD_CONST(Func, outputs)
     HALIDE_FORWARD_METHOD(Func, parallel)