mirror of
https://github.com/nod-ai/SHARK-Studio.git
synced 2026-01-11 06:48:09 -05:00
27 lines
762 B
Python
27 lines
762 B
Python
def pytest_addoption(parser):
|
|
# Attaches SHARK command-line arguments to the pytest machinery.
|
|
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(
|
|
"--benchmark",
|
|
action="store_true",
|
|
default="False",
|
|
help="Pass option to benchmark and write results.csv",
|
|
)
|
|
parser.addoption(
|
|
"--save_temps",
|
|
action="store_true",
|
|
default="False",
|
|
help="Saves IREE reproduction artifacts for filing upstream issues.",
|
|
)
|