zxybazh commented on code in PR #14624: URL: https://github.com/apache/tvm/pull/14624#discussion_r1175599526
########## include/tvm/relax/transform.h: ########## @@ -481,6 +481,18 @@ TVM_DLL Pass DeadCodeElimination(Array<runtime::String> entry_functions); */ TVM_DLL Pass ToMixedPrecision(const DataType& out_dtype); +/*! + * \brief The pass is designed for few shot tuning for static shape PrimFuncs. It examines all the + * blocks within the PrimFunc and conducts loop fusion, splitting, and other transformations based + * on MetaSchedule schedule rules but directly samples from the search space instead of using the + * tuning algorithm. User can specify the number of valid counts to try and whether to use runner + * for evaluation. + * \param valid_count The number of valid counts to try. + * \param runner The runner to evaluate the generated schedules. + * \return The Pass. + */ +TVM_DLL Pass FewShotTuning(Integer valid_count, ObjectRef runner); Review Comment: Yeah sure, will do! -- 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]
