ane-generation.yml

name: ANE Model Generation

on:
  push:
    branches: [macos-buildhwx]
  workflow_dispatch: # Allows manual triggering from the Actions tab

jobs:
  generate-and-convert:
    runs-on: macos-14
    steps:
      - uses: actions/checkout@v4

      - name: Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '3.10'

      - name: Install Dependencies
        run: |
          pip install numpy coremltools

      - name: Generate MLModel
        run: |
          python3 gen_mlmodel.py

      - name: Run coreml2hwx
        run: |
          git clone https://github.com/freedomtan/coreml_to_ane_hwx && cd coreml_to_ane_hwx
          make
          chmod +x ./coreml2hwx
          ./coreml2hwx ../test.mlmodel

      - name: Upload HWX Artifact
        uses: actions/upload-artifact@v4
        with:
          name: hwx-model
          path: /tmp/hwx_output/test/model.hwx