Documentation | Installation | Tutorials | Examples
This repository includes the code for AutoSA. AutoSA is an end-to-end systolic array compiler based on the polyhedral model. It takes algorithms in high-level programming languages (C) as inputs, performs polyhedral transformation and other architecture optimizations to map algorithms to systolic array architecture. The generated designs are in HLS C.
We offer a Docker image for quick start.
docker pull whbldhwj/autosa:latest
Let's try one small example. The input code can be found at ${AUTOSA_ROOT}/autosa_tests/mm/kernel.c
. The code region to be transformed to systolic array is annotated using a pair of pragmas scop
and endscop
.
Run the following command to compile generate a systolic array.
./autosa ./autosa_tests/mm/kernel.c \
--config=./autosa_config/autosa_config.json \
--target=autosa_hls_c \
--output-dir=./autosa.tmp/output \
--sa-sizes="{kernel[]->space_time[3];kernel[]->array_part[16,16,16];kernel[]->latency[8,8];kernel[]->simd[2]}" \
--simd-info=./autosa_tests/mm/simd_info.json \
--host-serialize
The generated code can be found in ${AUTOSA_ROOT}/autosa.tmp/output/src/
.
For detailed explaination of each AutoSA compilation option, please run
./autosa --help
or refer to AutoSA Compilation Options.
To generate the final bitsteam, set up your local Vitis development kit first. Then execute the makefile to build the design.
cp ${AUTOSA_ROOT}/autosa_tests/mm/Makefile autosa.tmp/output/
cp ${AUTOSA_ROOT}/autosa_tests/mm/connectivity.cfg autosa.tmp/output/
cd ${AUTOSA_ROOT}/autosa.tmp/output
make all
Makefile Options Descriptions
MODE := hw_emu
: Set the build configuration mode to HW Emulation, other modes: sw_emu|hwPLATFORM := xilinx_u250_xdma_201830_2
: Select the target platformKERNEL_SRC := src/kernel_kernel.cpp
: List the kernel source filesHOST_SRC := src/kernel_host.cpp
: List the host source filesThe connectivity.cfg
describes the DRAM port mapping. For more details about how to change the DRAM port mapping, please refer to the Xilinx tutorials.
AutoSA also supports generate HLS projects. Add the flag
--hls
to the command when compiling the program.
./autosa ./autosa_tests/mm/kernel.c \
--config=./autosa_config/autosa_config.json \
--target=autosa_hls_c \
--output-dir=./autosa.tmp/output \
--sa-sizes="{kernel[]->space_time[3];kernel[]->array_part[16,16,16];kernel[]->latency[8,8];kernel[]->simd[2]}" \
--simd-info=./autosa_tests/mm/simd_info.json \
--host-serialize \
--hls
AutoSA will generate an HLS host file ${AUTOSA_ROOT}/autosa.tmp/output/src/kernel_host.cpp
instead of the OpenCL host file generated in the previous step. To build the HLS project, run the following commands.
cp ${AUTOSA_ROOT}/autosa_scripts/hls_scripts/hls_script.tcl autosa.tmp/output/
cd ${AUTOSA_ROOT}/autosa.tmp/output
vivado_hls -f hls_script.tcl
For more detailed instructions on using AutoSA, please refer to the AutoSA Documentation.
AutoSA is open source for research purposes, and we would like to continously improve it! Please let us know if...
AutoSA is currently maintained by Jie Wang. Besides, we gratefully acknowledge the authors of PPCG for developing and actively maintaining PPCG as an open-source project.
More implementation details of AutoSA are covered in our paper. If you find this project useful in your research, please consider citing:
@inproceedings{wang2021autosa,
title={AutoSA: A Polyhedral Compiler for High-Performance Systolic Arrays on FPGA},
author={Wang, Jie and Guo, Licheng and Cong, Jason},
booktitle={Proceedings of the 2021 ACM/SIGDA International Symposium on Field-Programmable Gate Arrays},
year={2021}
}