wu-arch/kernels
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'ae' into ae-flash-virgo

Browse Source
This commit is contained in:
Hansung Kim
2025-01-30 01:49:05 -08:00
parent 96500e0abc 2c1ac4e938
commit c75ed0d531
4 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
kernels/flash_attention/compile_flash.sh
View File

@@ -27,6 +27,7 @@ ln -sf input.c.rand.fp32.seqlen1024headdim64.row.bin input.c.bin
for arch in "${archs[@]}"; do for arch in "${archs[@]}"; do
git checkout ae-flash-$arch git checkout ae-flash-$arch
git pull
# re-compile libvortexrt.a # re-compile libvortexrt.a
# FIXME after restructure # FIXME after restructure

4
kernels/sgemm_gemmini_dma/compile_virgo.sh
View File

@@ -1,5 +1,9 @@
#!/bin/sh #!/bin/sh
# hopper and virgo has the same SIMT configurations
git checkout ae-hopper
git pull
if [ ! -f input.a.rand01.fp16.m256n256k256.row.bin ]; then if [ ! -f input.a.rand01.fp16.m256n256k256.row.bin ]; then
echo "input binaries not found, generating operands" echo "input binaries not found, generating operands"
python3 generate_operands.py python3 generate_operands.py

2
kernels/sgemm_gemmini_dma/kernel.cpp
View File

@@ -53,7 +53,7 @@
#define PRINTF(...) sprintf(PRINT_BUF, __VA_ARGS__) #define PRINTF(...) sprintf(PRINT_BUF, __VA_ARGS__)
// #define PRINTF(...) vx_printf(__VA_ARGS__) // #define PRINTF(...) vx_printf(__VA_ARGS__)
#define SWISH(beta, x) ((x) / (1 + exp(-(beta) * (x)))) #define SWISH(beta, x) ((x) / (1 + exp(-(beta) * (x))))
// #define POWER #define POWER
typedef uint16_t smem_elem_t; typedef uint16_t smem_elem_t;
// typedef float smem_elem_t; // typedef float smem_elem_t;

1
kernels/sgemm_tcore/compile_tcore.sh
View File

@@ -53,6 +53,7 @@ done
for arch in "${archs[@]}"; do for arch in "${archs[@]}"; do
git checkout ae-$arch git checkout ae-$arch
git pull
# re-compile libvortexrt.a # re-compile libvortexrt.a
# FIXME after restructure # FIXME after restructure