junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r989378918
##
python/tvm/meta_schedule/relay_integration.py:
##
@@ -15,28 +15,101 @@
# specific language governing permissions and limitations
# under the License.
"""MetaSchedu
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r989280921
##
python/tvm/meta_schedule/relay_integration.py:
##
@@ -15,28 +15,101 @@
# specific language governing permissions and limitations
# under the License.
"""MetaSchedu
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r988422820
##
python/tvm/meta_schedule/relay_integration.py:
##
@@ -69,47 +124,229 @@ def extract_task_from_relay(
"""
# pylint: disable=import-outside-toplevel
from
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r988367063
##
python/tvm/meta_schedule/schedule_rule/schedule_rule.py:
##
@@ -76,6 +81,35 @@ def clone(self) -> "ScheduleRule":
"""
return _ffi_api.ScheduleRuleCl
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r988351862
##
src/meta_schedule/mutator/mutator.cc:
##
@@ -72,6 +97,10 @@ TVM_REGISTER_GLOBAL("meta_schedule.MutatorApply")
});
TVM_REGISTER_GLOBAL("meta_schedule.MutatorClo
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r988351447
##
src/meta_schedule/postproc/postproc.cc:
##
@@ -67,6 +109,10 @@
TVM_REGISTER_GLOBAL("meta_schedule.PostprocInitializeWithTuneContext")
TVM_REGISTER_GLOBAL("meta_sch
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r988351184
##
python/tvm/meta_schedule/postproc/postproc.py:
##
@@ -70,6 +74,35 @@ def clone(self) -> "Postproc":
"""
return _ffi_api.PostprocClone(self) # type:
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r988350143
##
src/meta_schedule/postproc/postproc.cc:
##
@@ -50,6 +50,48 @@ Postproc Postproc::PyPostproc(
return Postproc(n);
}
+Array Postproc::DefaultLLVM() {
+ return Ar
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r988344383
##
python/tvm/meta_schedule/database/database.py:
##
@@ -361,6 +363,44 @@ def current() -> Optional["Database"]:
"""Get the current database under scope."""
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r988341530
##
python/tvm/meta_schedule/cost_model/cost_model.py:
##
@@ -97,6 +104,30 @@ def predict(self, context: TuneContext, candidates:
List[MeasureCandidate]) -> n
)
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r988334549
##
python/tvm/meta_schedule/builder/builder.py:
##
@@ -126,6 +128,22 @@ def build(self, build_inputs: List[BuilderInput]) ->
List[BuilderResult]:
"""
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r988321661
##
python/tvm/contrib/torch/optimize_torch.py:
##
@@ -62,75 +51,32 @@ def forward(self, *torch_inputs: Tuple[torch.Tensor]):
return ret
-def llvm_target():
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r98832
##
python/tvm/contrib/torch/optimize_torch.py:
##
@@ -62,75 +51,32 @@ def forward(self, *torch_inputs: Tuple[torch.Tensor]):
return ret
-def llvm_target():
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r985899276
##
python/tvm/meta_schedule/relay_integration.py:
##
@@ -15,28 +15,82 @@
# specific language governing permissions and limitations
# under the License.
"""MetaSchedul
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r985387124
##
python/tvm/meta_schedule/task_scheduler/task_scheduler.py:
##
@@ -101,15 +90,43 @@ def join_running_task(self, task_id: int) ->
List[RunnerResult]:
"""
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r985191319
##
include/tvm/meta_schedule/search_strategy.h:
##
@@ -88,14 +88,17 @@ class SearchStrategyNode : public runtime::Object {
/*!
* \brief Pre-tuning for the searc
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r985191082
##
src/meta_schedule/task_scheduler/task_scheduler.cc:
##
@@ -21,83 +21,225 @@
namespace tvm {
namespace meta_schedule {
-void TaskSchedulerNode::InitializeTask(int
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r985190975
##
python/tvm/meta_schedule/relay_integration.py:
##
@@ -15,28 +15,82 @@
# specific language governing permissions and limitations
# under the License.
"""MetaSchedul
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. *
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r985190584
##
include/tvm/meta_schedule/task_scheduler.h:
##
@@ -73,66 +127,71 @@ namespace meta_schedule {
*/
class TaskSchedulerNode : public runtime::Object {
public:
- /*!
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r985190468
##
include/tvm/meta_schedule/space_generator.h:
##
@@ -171,6 +213,7 @@ class PySpaceGeneratorNode : public SpaceGeneratorNode {
FClone f_clone;
void VisitAttrs(t
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r985190427
##
include/tvm/meta_schedule/search_strategy.h:
##
@@ -257,8 +249,8 @@ class PySearchStrategyNode : public SearchStrategyNode {
}
void InitializeWithTuneContext(
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r985190272
##
python/tvm/meta_schedule/logging.py:
##
@@ -0,0 +1,262 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. Se
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r985190154
##
python/tvm/meta_schedule/utils.py:
##
@@ -114,7 +113,10 @@ def __init__(self, *args, **kwargs):
def __getattr__(self, name: str):
"""Bridge th
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r985051195
##
tests/python/unittest/test_meta_schedule_space_generator.py:
##
@@ -104,4 +108,5 @@ class TestPySpaceGenerator(PySpaceGenerator):
if __name__ == "__main__":
-
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r985051216
##
include/tvm/support/random_engine.h:
##
@@ -46,32 +43,18 @@ namespace support {
class LinearCongruentialEngine {
public:
- /*!
- * \brief The result type is d
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r985051150
##
src/meta_schedule/schedule_rule/schedule_rule.cc:
##
@@ -51,6 +51,125 @@ ScheduleRule ScheduleRule::PyScheduleRule(
return ScheduleRule(n);
}
+Array ScheduleRul
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r983980554
##
python/tvm/meta_schedule/utils.py:
##
@@ -114,7 +113,10 @@ def __init__(self, *args, **kwargs):
def __getattr__(self, name: str):
"""Bridge th
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r982961786
##
src/meta_schedule/schedule_rule/schedule_rule.cc:
##
@@ -51,6 +51,125 @@ ScheduleRule ScheduleRule::PyScheduleRule(
return ScheduleRule(n);
}
+Array ScheduleRul
junrushao commented on code in PR #12895:
URL: https://github.com/apache/tvm/pull/12895#discussion_r982931215
##
src/meta_schedule/schedule_rule/schedule_rule.cc:
##
@@ -51,6 +51,125 @@ ScheduleRule ScheduleRule::PyScheduleRule(
return ScheduleRule(n);
}
+Array ScheduleRul
30 matches
Mail list logo
