junrushao1994 commented on code in PR #11683:
URL: https://github.com/apache/tvm/pull/11683#discussion_r899901175
##########
python/tvm/meta_schedule/tune_context.py:
##########
@@ -287,3 +287,64 @@ def notify_runner_results(
"Please construct TuneContext with search_strategy"
)
return self.search_strategy.notify_runner_results(measure_candidates,
results)
+
+ def get_measure_candidate(self, mod, trace):
+ """Generate a measure candidate given an initial IR module and a trace.
+
+ Parameters
+ -----------
+ mod : IRModule
+ The initial IR module.
+ trace : Trace
+ The trace applying to the IR Module.
+
+ Returns
+ -------
+ candidate : MeasureCandidate
+ A generated candidate.
+ """
+ return _ffi_api.TuneContextGetMeasureCandidate(self, mod, trace) #
type: ignore # pylint: disable=no-member
+
+ def set_measure_candidates(self, candidates):
Review Comment:
I'm not super opinionated, but given those are private APIs in C++, shall we
add an underscore prefix to them to be consistent with C++ side?
--
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]
