[Easy][Dynamo][TVM] remove unnecessary prints (#142445)

This PR intends to remove the unnecessary prints in the auto-scheduler of dynamo's TVM backend.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/142445
Approved by: https://github.com/jansel
This commit is contained in:
Yuanjing Shi 2024-12-10 19:52:00 +00:00 committed by PyTorch MergeBot
parent e95bd337e1
commit 117b6c3e2c

View File

@ -63,10 +63,6 @@ def tvm(
tasks, task_weights = auto_scheduler.extract_tasks(
mod["main"], params, target
)
for task in tasks:
print(task.compute_dag)
else:
print("No tasks")
if len(tasks) != 0:
tuner = auto_scheduler.TaskScheduler(tasks, task_weights)
if not os.path.exists(log_file):