diff options
Diffstat (limited to 'compute_kernel_writer/prototype/src')
| -rw-r--r-- | compute_kernel_writer/prototype/src/KernelWriter.cpp | 5 | ||||
| -rw-r--r-- | compute_kernel_writer/prototype/src/Prototype.h | 10 |
2 files changed, 7 insertions, 8 deletions
diff --git a/compute_kernel_writer/prototype/src/KernelWriter.cpp b/compute_kernel_writer/prototype/src/KernelWriter.cpp index 1ac9ede5b5..9122e518b4 100644 --- a/compute_kernel_writer/prototype/src/KernelWriter.cpp +++ b/compute_kernel_writer/prototype/src/KernelWriter.cpp @@ -270,13 +270,14 @@ void KernelWriter::op_else(const std::function<void()> &body) _impl->compound_statement_end(); } -void KernelWriter::op_for_loop(const TileOperand &var_name, BinaryOp cond_op, const TileOperand &cond_value_name, AssignmentOp update_op, const TileOperand &update_value_name, const std::function<void()> &body) +void KernelWriter::op_for_loop(const TileOperand &var_name, BinaryOp cond_op, const TileOperand &cond_value_name, const TileOperand &update_var_name, AssignmentOp update_op, const TileOperand &update_value_name, const std::function<void()> &body) { auto impl_var_name = var_name.create_impl_operand(_impl.get()); auto impl_cond_value_name = cond_value_name.create_impl_operand(_impl.get()); + auto impl_update_var_name = update_var_name.create_impl_operand(_impl.get()); auto impl_update_value_name = update_value_name.create_impl_operand(_impl.get()); - _impl->op_for_loop_header(impl_var_name, cond_op, impl_cond_value_name, update_op, impl_update_value_name); + _impl->op_for_loop_header(impl_var_name, cond_op, impl_cond_value_name, impl_update_var_name, update_op, impl_update_value_name); _impl->compound_statement_begin(); body(); _impl->compound_statement_end(); diff --git a/compute_kernel_writer/prototype/src/Prototype.h b/compute_kernel_writer/prototype/src/Prototype.h index 72fa419fc2..05c7306e3a 100644 --- a/compute_kernel_writer/prototype/src/Prototype.h +++ b/compute_kernel_writer/prototype/src/Prototype.h @@ -2498,7 +2498,7 @@ public: virtual void op_else_header() = 0; - virtual void op_for_loop_header(const Operand &var_name, BinaryOp cond_op, const Operand &cond_value, AssignmentOp update_op, const Operand &update_value) = 0; + virtual void op_for_loop_header(const Operand &var_name, BinaryOp cond_op, const Operand &cond_value, const Operand &update_var, AssignmentOp update_op, const Operand &update_value) = 0; virtual void op_load_indirect(const TensorOperand &tensor, const Operand &dst, const Operand &x, const Operand &y_indirect, const Operand &z, const Operand &b = Operand("0", OperandType::ScalarInt32)) = 0; @@ -3654,9 +3654,6 @@ public: case UnaryFunction::Fabs: _data->code += "fabs("; break; - case UnaryFunction::IsGreaterEqual: - _data->code += "isgreaterequal("; - break; case UnaryFunction::Log: _data->code += "log("; break; @@ -3798,11 +3795,12 @@ public: _data->code += "else\n"; } - void op_for_loop_header(const Operand& var_name, BinaryOp cond_op, const Operand& cond_value_name, AssignmentOp update_op, const Operand& update_value_name) override + void op_for_loop_header(const Operand& var_name, BinaryOp cond_op, const Operand& cond_value_name, const Operand &update_var_name, AssignmentOp update_op, const Operand& update_value_name) override { OperandUnpacker operands(_data->tiles, _data->arguments); const IVectorTile *var = operands.unpack(var_name); const IVectorTile *cond_value = operands.unpack(cond_value_name); + const IVectorTile *update_var = operands.unpack(update_var_name); const IVectorTile *update_value = operands.unpack(update_value_name); const int32_t dst_w = var->format().w; @@ -3818,7 +3816,7 @@ public: _data->code += " "; _data->code += to_string(cond_op); _data->code += " " + cond_value->scalar(0, 0).str + "; "; - _data->code += var->scalar(0, 0).str; + _data->code += update_var->scalar(0, 0).str; _data->code += " "; _data->code += to_string(update_op); _data->code += " " + update_value->scalar(0, 0).str + ")"; |