diff options
Diffstat (limited to 'src/dynamic_fusion/sketch/gpu/GpuOperatorGroup.h')
-rw-r--r-- | src/dynamic_fusion/sketch/gpu/GpuOperatorGroup.h | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/src/dynamic_fusion/sketch/gpu/GpuOperatorGroup.h b/src/dynamic_fusion/sketch/gpu/GpuOperatorGroup.h index 35abe6c543..308a9d796a 100644 --- a/src/dynamic_fusion/sketch/gpu/GpuOperatorGroup.h +++ b/src/dynamic_fusion/sketch/gpu/GpuOperatorGroup.h @@ -77,17 +77,19 @@ public: static constexpr size_t max_fused_operators = 32; /** Try adding (without actually adding) an operator to the group * - * @param[in] op Operator to be added + * @param[in] op Operator to be added + * @param[in] is_output Whether this operator is the output operator. * * @return true If @p op can be added while maintaining the invariants * @return false Otherwise */ - bool try_add_operator(const Operator &op) const; + bool try_add_operator(const Operator &op, bool is_output = false) const; /** Add an operator to the group * - * @param[in] op Operator to be added + * @param[in] op Operator to be added + * @param[in] is_output Whether this operator is the output operator. */ - void add_operator(const Operator &op); + void add_operator(const Operator &op, bool is_output = false); /** Create a new operator * * @param[in] operator_type @ref GpuOperatorType of the new operator |