Skip to content

Commit

Permalink
Remove jit from the framework code (#1787)
Browse files Browse the repository at this point in the history
Summary:
Jit(torchscript) is now in archive mode and there is no need that we add it to the explicit options.

Jit(torchscript) is still supported with the extra args option `--backend torchsript`.

Fixes #1777

Pull Request resolved: #1787

Reviewed By: msaroufim

Differential Revision: D47774649

Pulled By: xuzhao9

fbshipit-source-id: 7eb25e54d016da9a49b93c514cc302f6d8d4573c
  • Loading branch information
xuzhao9 authored and facebook-github-bot committed Jul 26, 2023
1 parent 770d5cf commit 4ea9354
Show file tree
Hide file tree
Showing 198 changed files with 287 additions and 791 deletions.
45 changes: 0 additions & 45 deletions compare.py

This file was deleted.

5 changes: 0 additions & 5 deletions compare.sh

This file was deleted.

3 changes: 0 additions & 3 deletions conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,9 +11,6 @@ def pytest_addoption(parser):
help="Disable checks/assertions for machine configuration for stable benchmarks")
parser.addoption("--disable_nograd", action='store_true',
help="Disable no_grad for eval() runs")
parser.addoption("--check_opt_vs_noopt_jit",
action='store_true',
help="The best attempt to check results for inference runs. Not all models support this!")
parser.addoption("--cpu_only", action='store_true',
help="Run benchmarks on cpu only and ignore machine configuration checks")
parser.addoption("--cuda_only", action='store_true',
Expand Down
237 changes: 0 additions & 237 deletions fx_profile.py

This file was deleted.

4 changes: 2 additions & 2 deletions gen_summary_metadata.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ def _extract_detail(path: str) -> Dict[str, Any]:
# Separate train and eval to isolated processes.
task_t = ModelTask(path, timeout=TIMEOUT)
try:
task_t.make_model_instance(device=device, jit=False)
task_t.make_model_instance(device=device)
task_t.set_train()
task_t.train()
task_t.extract_details_train()
Expand All @@ -64,7 +64,7 @@ def _extract_detail(path: str) -> Dict[str, Any]:

task_e = ModelTask(path, timeout=TIMEOUT)
try:
task_e.make_model_instance(device=device, jit=False)
task_e.make_model_instance(device=device)
task_e.set_eval()
task_e.eval()
task_e.extract_details_eval()
Expand Down
8 changes: 0 additions & 8 deletions run.py
Original file line number Diff line number Diff line change
Expand Up @@ -293,13 +293,6 @@ def _validate_profile_options(profile_options: str):
default="cpu",
help="Which device to use.",
)
parser.add_argument(
"-m",
"--mode",
choices=["eager", "jit"],
default="eager",
help="Which mode to run.",
)
parser.add_argument(
"-t",
"--test",
Expand Down Expand Up @@ -407,7 +400,6 @@ def _validate_profile_options(profile_options: str):
m = Model(
device=args.device,
test=args.test,
jit=(args.mode == "jit"),
batch_size=args.bs,
extra_args=extra_args,
)
Expand Down
Loading

0 comments on commit 4ea9354

Please sign in to comment.