Skip to content

Commit

Permalink
remove jit from dynamo benchmark (pytorch#113338)
Browse files Browse the repository at this point in the history
Continuous of pytorch#106071, without this dynamo dist cannot run at the moment.

Related to pytorch/benchmark#1787

Pull Request resolved: pytorch#113338
Approved by: https://github.com/ezyang
  • Loading branch information
oraluben authored and pytorchmergebot committed Nov 10, 2023
1 parent 2cd8c05 commit b1eb9e1
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions benchmarks/dynamo/dist_util.py
Original file line number Diff line number Diff line change
Expand Up @@ -94,9 +94,7 @@ def get_model(args):
f"torchbenchmark.models.{args.torchbench_model}"
)
benchmark_cls = getattr(module, "Model", None)
bm = benchmark_cls(
test="train", device=args.device, jit=False, batch_size=args.batch_size
)
bm = benchmark_cls(test="train", device=args.device, batch_size=args.batch_size)
model, inputs = bm.get_module()
elif args.toy_model:
model = ToyModel()
Expand Down

0 comments on commit b1eb9e1

Please sign in to comment.