diff --git a/api/steps/workload/01_build_event.step.py b/api/steps/workload/01_build_event.step.py index a3eec81..2662d23 100644 --- a/api/steps/workload/01_build_event.step.py +++ b/api/steps/workload/01_build_event.step.py @@ -40,6 +40,11 @@ "buddynext": "BuddyNext", } +# CLI model → -DMODEL cmake value for workload builds. +MODEL_CMAKE = { + "alexnet": "alexnet", +} + def chips_for_model(bbdir: str, model_key: str) -> set[str]: """Chips that currently ship a layout for this model (many-to-many)."""