├── .gitignore ├── README.md ├── autotuning.sh ├── copy_remote_back.sh ├── copy_to_remote.sh ├── doc └── Triton kernel performance on RISC-V CPU.pdf ├── include ├── kernel │ ├── correlation.h │ ├── dropout.h │ ├── layernorm.h │ ├── matmul.h │ ├── resize.h │ ├── rope.h │ ├── softmax.h │ └── warp.h └── support │ ├── benchmark.h │ ├── omp.h │ └── support.h ├── patch ├── sysroot.tar.xz ├── triton-cpu-0001-RISCV.patch ├── triton-cpu-0002-Autotuning.patch └── triton-cpu-0003-gather.patch ├── plot_autotuning.py ├── plot_benchmark.py ├── report.sh ├── run.sh ├── run_local.sh └── src ├── c ├── correlation.cpp ├── dropout.cpp ├── layernorm.cpp ├── matmul.cpp ├── resize.cpp ├── rope.cpp ├── softmax.cpp └── warp.cpp ├── main ├── correlation.cfg ├── correlation.cpp ├── dropout.cfg ├── dropout.cpp ├── layernorm.cfg ├── layernorm.cpp ├── matmul.cfg ├── matmul.cpp ├── resize.cfg ├── resize.cpp ├── rope.cfg ├── rope.cpp ├── softmax_kernel.cfg ├── softmax_kernel.cpp ├── warp.cfg └── warp.cpp ├── support └── support.cpp └── triton ├── correlation.py ├── dropout.py ├── layernorm.py ├── matmul.py ├── resize.py ├── rope.py ├── softmax.py └── warp.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/README.md -------------------------------------------------------------------------------- /autotuning.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/autotuning.sh -------------------------------------------------------------------------------- /copy_remote_back.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/copy_remote_back.sh -------------------------------------------------------------------------------- /copy_to_remote.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/copy_to_remote.sh -------------------------------------------------------------------------------- /doc/Triton kernel performance on RISC-V CPU.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/doc/Triton kernel performance on RISC-V CPU.pdf -------------------------------------------------------------------------------- /include/kernel/correlation.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/include/kernel/correlation.h -------------------------------------------------------------------------------- /include/kernel/dropout.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/include/kernel/dropout.h -------------------------------------------------------------------------------- /include/kernel/layernorm.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/include/kernel/layernorm.h -------------------------------------------------------------------------------- /include/kernel/matmul.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/include/kernel/matmul.h -------------------------------------------------------------------------------- /include/kernel/resize.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/include/kernel/resize.h -------------------------------------------------------------------------------- /include/kernel/rope.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/include/kernel/rope.h -------------------------------------------------------------------------------- /include/kernel/softmax.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/include/kernel/softmax.h -------------------------------------------------------------------------------- /include/kernel/warp.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/include/kernel/warp.h -------------------------------------------------------------------------------- /include/support/benchmark.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/include/support/benchmark.h -------------------------------------------------------------------------------- /include/support/omp.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/include/support/omp.h -------------------------------------------------------------------------------- /include/support/support.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/include/support/support.h -------------------------------------------------------------------------------- /patch/sysroot.tar.xz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/patch/sysroot.tar.xz -------------------------------------------------------------------------------- /patch/triton-cpu-0001-RISCV.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/patch/triton-cpu-0001-RISCV.patch -------------------------------------------------------------------------------- /patch/triton-cpu-0002-Autotuning.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/patch/triton-cpu-0002-Autotuning.patch -------------------------------------------------------------------------------- /patch/triton-cpu-0003-gather.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/patch/triton-cpu-0003-gather.patch -------------------------------------------------------------------------------- /plot_autotuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/plot_autotuning.py -------------------------------------------------------------------------------- /plot_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/plot_benchmark.py -------------------------------------------------------------------------------- /report.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/report.sh -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/run.sh -------------------------------------------------------------------------------- /run_local.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/run_local.sh -------------------------------------------------------------------------------- /src/c/correlation.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/c/correlation.cpp -------------------------------------------------------------------------------- /src/c/dropout.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/c/dropout.cpp -------------------------------------------------------------------------------- /src/c/layernorm.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/c/layernorm.cpp -------------------------------------------------------------------------------- /src/c/matmul.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/c/matmul.cpp -------------------------------------------------------------------------------- /src/c/resize.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/c/resize.cpp -------------------------------------------------------------------------------- /src/c/rope.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/c/rope.cpp -------------------------------------------------------------------------------- /src/c/softmax.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/c/softmax.cpp -------------------------------------------------------------------------------- /src/c/warp.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/c/warp.cpp -------------------------------------------------------------------------------- /src/main/correlation.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/main/correlation.cfg -------------------------------------------------------------------------------- /src/main/correlation.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/main/correlation.cpp -------------------------------------------------------------------------------- /src/main/dropout.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/main/dropout.cfg -------------------------------------------------------------------------------- /src/main/dropout.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/main/dropout.cpp -------------------------------------------------------------------------------- /src/main/layernorm.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/main/layernorm.cfg -------------------------------------------------------------------------------- /src/main/layernorm.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/main/layernorm.cpp -------------------------------------------------------------------------------- /src/main/matmul.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/main/matmul.cfg -------------------------------------------------------------------------------- /src/main/matmul.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/main/matmul.cpp -------------------------------------------------------------------------------- /src/main/resize.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/main/resize.cfg -------------------------------------------------------------------------------- /src/main/resize.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/main/resize.cpp -------------------------------------------------------------------------------- /src/main/rope.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/main/rope.cfg -------------------------------------------------------------------------------- /src/main/rope.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/main/rope.cpp -------------------------------------------------------------------------------- /src/main/softmax_kernel.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/main/softmax_kernel.cfg -------------------------------------------------------------------------------- /src/main/softmax_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/main/softmax_kernel.cpp -------------------------------------------------------------------------------- /src/main/warp.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/main/warp.cfg -------------------------------------------------------------------------------- /src/main/warp.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/main/warp.cpp -------------------------------------------------------------------------------- /src/support/support.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/support/support.cpp -------------------------------------------------------------------------------- /src/triton/correlation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/triton/correlation.py -------------------------------------------------------------------------------- /src/triton/dropout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/triton/dropout.py -------------------------------------------------------------------------------- /src/triton/layernorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/triton/layernorm.py -------------------------------------------------------------------------------- /src/triton/matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/triton/matmul.py -------------------------------------------------------------------------------- /src/triton/resize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/triton/resize.py -------------------------------------------------------------------------------- /src/triton/rope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/triton/rope.py -------------------------------------------------------------------------------- /src/triton/softmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/triton/softmax.py -------------------------------------------------------------------------------- /src/triton/warp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Terapines/AI-Benchmark/HEAD/src/triton/warp.py --------------------------------------------------------------------------------