mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
37093f4d99
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/44291 Test Plan: Imported from OSS Reviewed By: bertmaher Differential Revision: D23569089 Pulled By: ZolotukhinM fbshipit-source-id: ec25b2f0bba303adaa46c3e85b1a9ce4fa3cf076
18 lines
891 B
Python
18 lines
891 B
Python
import pytest # noqa: F401
|
|
|
|
default_rnns = ['cudnn', 'aten', 'jit', 'jit_premul', 'jit_premul_bias', 'jit_simple',
|
|
'jit_multilayer', 'py']
|
|
default_cnns = ['resnet18', 'resnet18_jit', 'resnet50', 'resnet50_jit']
|
|
all_nets = default_rnns + default_cnns
|
|
|
|
def pytest_generate_tests(metafunc):
|
|
# This creates lists of tests to generate, can be customized
|
|
if metafunc.cls.__name__ == "TestBenchNetwork":
|
|
metafunc.parametrize('net_name', all_nets, scope="class")
|
|
metafunc.parametrize("executor", [metafunc.config.getoption("executor")], scope="class")
|
|
metafunc.parametrize("fuser", [metafunc.config.getoption("fuser")], scope="class")
|
|
|
|
def pytest_addoption(parser):
|
|
parser.addoption("--fuser", default="old", help="fuser to use for benchmarks")
|
|
parser.addoption("--executor", default="legacy", help="executor to use for benchmarks")
|