Lunderberg commented on code in PR #15103:
URL: https://github.com/apache/tvm/pull/15103#discussion_r1285217773
##########
src/target/source/codegen_c.h:
##########
@@ -65,12 +65,30 @@ class CodeGenC : public ExprFunctor<void(const PrimExpr&,
std::ostream&)>,
* \param output_ssa Whether output SSA.
*/
void Init(bool output_ssa);
+
/*!
* \brief Add the function to the generated module.
- * \param f The function to be compiled.
+ * \param gvar The GlobalVar representing the function.
+ * \param func The function to be compiled.
* \param whether to append return 0 in the end.
*/
- void AddFunction(const PrimFunc& f);
+ virtual void DeclareFunction(const GlobalVar& gvar, const PrimFunc& func);
+
+ /*!
+ * \brief Add the function to the generated module.
Review Comment:
Thank you, and updated.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
