Skip to content

Commit

Permalink
split thunderfx executor from default benchmark executors
Browse files Browse the repository at this point in the history
  • Loading branch information
riccardofelluga committed Oct 14, 2024
1 parent f840a35 commit 56b916c
Showing 1 changed file with 10 additions and 10 deletions.
20 changes: 10 additions & 10 deletions thunder/benchmarks/targets.py
Original file line number Diff line number Diff line change
Expand Up @@ -152,16 +152,16 @@ def interpreter_fwd(module: Callable):
return fn_


executors = (
torch_executor,
torch_compile_executor,
thunder_executor,
thunderfx_executor,
)
executors = (torch_executor, torch_compile_executor, thunder_executor)
executors_ids = (
"torch",
"torch.compile",
"thunder",
)

torchbench_executors = (*executors, thunderfx_executor)
torchbench_executors_ids = (
*executors_ids,
"thunderfx",
)

Expand Down Expand Up @@ -844,8 +844,8 @@ def test_resnet50(benchmark, executor: Callable, compute_type: ComputeType):
)
@pytest.mark.parametrize(
"executor,",
executors,
ids=executors_ids,
torchbench_executors,
ids=torchbench_executors_ids,
)
@parametrize_compute_type
def test_torchbench(benchmark, module_name, executor, compute_type: ComputeType):
Expand All @@ -870,8 +870,8 @@ def test_torchbench(benchmark, module_name, executor, compute_type: ComputeType)
)
@pytest.mark.parametrize(
"executor,",
executors,
ids=executors_ids,
torchbench_executors,
ids=torchbench_executors_ids,
)
@parametrize_compute_type
def test_torchbench_canary(benchmark, module_name, executor, compute_type: ComputeType):
Expand Down

0 comments on commit 56b916c

Please sign in to comment.