This is an automated email from the ASF dual-hosted git repository.
junrushao pushed a commit to branch unity-staging
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/unity-staging by this push:
new d8f79a45e0 fix
d8f79a45e0 is described below
commit d8f79a45e0e301a5cac64828423ee6a4949533cb
Author: Junru Shao <[email protected]>
AuthorDate: Sun May 7 10:07:26 2023 -0700
fix
---
python/tvm/meta_schedule/tir_integration.py | 4 ----
1 file changed, 4 deletions(-)
diff --git a/python/tvm/meta_schedule/tir_integration.py
b/python/tvm/meta_schedule/tir_integration.py
index 6d0af39923..36e89a11b4 100644
--- a/python/tvm/meta_schedule/tir_integration.py
+++ b/python/tvm/meta_schedule/tir_integration.py
@@ -175,7 +175,6 @@ def _tune_tir(
task_scheduler: TaskScheduler.TaskSchedulerType = "round-robin",
space: SpaceGenerator.SpaceGeneratorType = "post-order-apply",
strategy: SearchStrategy.SearchStrategyType = "evolutionary",
- task_name: str = "main",
num_tuning_cores: Union[Literal["physical", "logical"], int] = "physical",
seed: Optional[int] = None,
) -> Database:
@@ -209,8 +208,6 @@ def _tune_tir(
The space generator.
strategy : SearchStrategy.SearchStrategyType
The search strategy.
- task_name : str
- The name of the task.
num_tuning_cores : Union[Literal["physical", "logical"], int]
The number of CPU cores to use during tuning.
seed : Optional[int]
@@ -237,7 +234,6 @@ def _tune_tir(
task_scheduler=task_scheduler,
space=space,
strategy=strategy,
- task_name=task_name,
num_tuning_cores=num_tuning_cores,
seed=seed,
)
