Skip to content

feat: add AlexNet MODEL_CMAKE entry for workload builds - #13

Closed
shirohasuki wants to merge 1 commit into
mainfrom
alexnet-model-cmake
Closed

feat: add AlexNet MODEL_CMAKE entry for workload builds#13
shirohasuki wants to merge 1 commit into
mainfrom
alexnet-model-cmake

Conversation

@shirohasuki

Copy link
Copy Markdown
Member

Summary

Completes the three-place registration for the new AlexNet ModelTest e2e workload (Pie33000/alexnet): adds the CLI alias -> -DMODEL cmake value mapping "alexnet": "alexnet" to the MODEL_CMAKE dict in api/steps/workload/01_build_event.step.py.

(The workload itself and the two CMake registrations are delivered in buddy-compiler/buddy-examples#16.)

Verification

  • Local run device: CPU (x86-64), one forward ≈ 6.1 s
  • MLIR forward vs torch reference: max abs logits diff 9.5e-6, argmax/top-5 identical
  • buckyball_workload_audit: ACCEPT

Registers the CLI model alias 'alexnet' -> -DMODEL value 'alexnet' in the
MODEL_CMAKE dict of the workload build step, completing the three-place
registration for the new AlexNet ModelTest e2e workload
(https://huggingface.co/Pie33000/alexnet).
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant