Skip to content

Commit 9131695

Browse files
committed
chore: linter fixes
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent 3c92d82 commit 9131695

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

tools/perf/perf_run.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -616,10 +616,18 @@ def load_torch_model(params):
616616
)
617617

618618
if use_dynamo and (model_torch is None):
619-
raise ValueError("No Pytorch model (nn.Module) is provided for torchdynamo compilation. Please provide a pytorch model")
619+
raise ValueError(
620+
"No Pytorch model (nn.Module) is provided for torchdynamo compilation. Please provide a pytorch model"
621+
)
620622

621623
if use_dynamo and model_torch:
622-
model_torch = torch.compile(model_torch, "default", dynamic=False, fullgraph=False, backend=dynamo_backend)
624+
model_torch = torch.compile(
625+
model_torch,
626+
"default",
627+
dynamic=False,
628+
fullgraph=False,
629+
backend=dynamo_backend,
630+
)
623631

624632
backends = parse_backends(params["backends"])
625633
truncate_long_and_double = params["truncate"]

0 commit comments

Comments
 (0)