merge kernel changes from kernels-asplos-ae

This commit is contained in:
Richard Yan
2025-01-29 22:11:25 -08:00
parent a61bf257ff
commit 91a82c9f0f
24 changed files with 146 additions and 62 deletions

View File

@@ -0,0 +1,21 @@
#!/bin/sh
if [ ! -f input.a.rand01.fp16.m256n256k256.row.bin ]; then
echo "input binaries not found, generating operands"
python3 generate_operands.py
fi
for a in args/*; do
echo "compiling GEMM kernel for Virgo with dim ${a}"
cp -f $a args.bin
aa=$(basename "$a")
cp -f input.a.rand01.fp16.m${aa}n${aa}k${aa}.row.bin input.a.bin
cp -f input.b.rand01.fp16.m${aa}n${aa}k${aa}.row.bin input.b.bin
touch input.c.bin
# touch source file to force re-building, as the Makefile does not track
# binary changes
touch kernel.cpp
make CONFIG=gemm.virgo.hopper.dim${aa}
done