Skip to content

Commit a2e015a

Browse files
committed
ci: add Arm-hosted Graviton4 runner
1 parent 2f0c2db commit a2e015a

File tree

2 files changed

+30
-1
lines changed

2 files changed

+30
-1
lines changed

.github/workflows/build.yml

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1649,3 +1649,27 @@ jobs:
16491649
run: |
16501650
GG_BUILD_KLEIDIAI=1 GG_BUILD_EXTRA_TESTS_0=1 bash ./ci/run.sh ./tmp/results ./tmp/mnt
16511651
1652+
ggml-ci-arm64-graviton4-kleidiai:
1653+
runs-on: ah-ubuntu_22_04-c8g_8x
1654+
1655+
steps:
1656+
- name: Clone
1657+
id: checkout
1658+
uses: actions/checkout@v4
1659+
1660+
- name: ccache
1661+
uses: ggml-org/[email protected]
1662+
with:
1663+
key: ggml-ci-arm64-graviton4-kleidiai
1664+
evict-old-files: 1d
1665+
1666+
- name: Dependencies
1667+
id: depends
1668+
run: |
1669+
sudo apt-get update
1670+
sudo apt-get install -y build-essential libcurl4-openssl-dev
1671+
1672+
- name: Test
1673+
id: ggml-ci
1674+
run: |
1675+
GG_BUILD_KLEIDIAI=1 GG_BUILD_EXTRA_TESTS_0=1 bash ./ci/run.sh ./tmp/results ./tmp/mnt

ci/run.sh

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,12 @@ fi
121121
if [ -n "${GG_BUILD_KLEIDIAI}" ]; then
122122
echo ">>===== Enabling KleidiAI support"
123123

124-
CANDIDATES=("armv9-a+dotprod+i8mm" "armv8.6-a+dotprod+i8mm" "armv8.2-a+dotprod")
124+
CANDIDATES=(
125+
"armv9-a+dotprod+i8mm+sve2"
126+
"armv9-a+dotprod+i8mm"
127+
"armv8.6-a+dotprod+i8mm"
128+
"armv8.2-a+dotprod"
129+
)
125130
CPU=""
126131

127132
for cpu in "${CANDIDATES[@]}"; do

0 commit comments

Comments
 (0)