mirror of
https://github.com/nod-ai/SHARK-Studio.git
synced 2026-01-10 06:17:55 -05:00
3514822cac
* Add ONNX env var flags for venv setup. * Setup arguments for ONNX benchmarking via pytest. * Enable ONNX benchmarking on MiniLM via pytest (experimental) * Fix sequence lengths to 128 for TF model creation and fix issue with benchmarks. * Disable CI CPU benchmarks on A100, change some default args. * add xfails for roberta TF model tests on GPU.
34 lines
997 B
Python
34 lines
997 B
Python
def pytest_addoption(parser):
|
|
# Attaches SHARK command-line arguments to the pytest machinery.
|
|
parser.addoption(
|
|
"--benchmark",
|
|
action="store_true",
|
|
default="False",
|
|
help="Pass option to benchmark and write results.csv",
|
|
)
|
|
parser.addoption(
|
|
"--onnx_bench",
|
|
action="store_true",
|
|
default="False",
|
|
help="Add ONNX benchmark results to pytest benchmarks.",
|
|
)
|
|
# The following options are deprecated and pending removal.
|
|
parser.addoption(
|
|
"--save_mlir",
|
|
action="store_true",
|
|
default="False",
|
|
help="Pass option to save input MLIR",
|
|
)
|
|
parser.addoption(
|
|
"--save_vmfb",
|
|
action="store_true",
|
|
default="False",
|
|
help="Pass option to save IREE output .vmfb",
|
|
)
|
|
parser.addoption(
|
|
"--save_temps",
|
|
action="store_true",
|
|
default="False",
|
|
help="Saves IREE reproduction artifacts for filing upstream issues.",
|
|
)
|