mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
[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:
parent
e95bd337e1
commit
117b6c3e2c
|
|
@ -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):
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user