mirror of
https://github.com/nod-ai/AMD-SHARK-Studio.git
synced 2026-04-03 03:00:17 -04:00
19 lines
750 B
Python
19 lines
750 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.")
|