junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r985190735
##########
include/tvm/meta_schedule/task_scheduler.h:
##########
@@ -143,55 +202,43 @@ class TaskScheduler;
/*! \brief The task scheduler with customized methods on the python-side. */
class PyTaskSchedulerNode : public TaskSchedulerNode {
public:
- /*! \brief The function type of `Tune` method. */
- using FTune = runtime::TypedPackedFunc<void()>;
-
- /*! \brief The function type of `InitializeTask` method. */
- using FInitializeTask = runtime::TypedPackedFunc<void(int)>;
-
/*!
- * \brief The function type of `TouchTask` method.
- * \param task_id The task id to be checked.
- * \return Whether the task is running.
+ * \brief The function type of `NextTaskId` method.
+ * \return The next task id.
*/
- using FTouchTask = runtime::TypedPackedFunc<void(int)>;
-
+ using FNextTaskId = runtime::TypedPackedFunc<int()>;
/*!
* \brief The function type of `JoinRunningTask` method.
* \param task_id The task id to be joined.
*/
using FJoinRunningTask = runtime::TypedPackedFunc<Array<RunnerResult>(int)>;
+ /*! \brief The function type of `Tune` method. */
+ using FTune = runtime::TypedPackedFunc<void(
Review Comment:
done
--
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]
