wu-arch/kernels

Files

History

Hansung Kim f7603b18d3 flash.py: Write V to file

2024-09-01 18:17:05 -07:00

..

check_correctness.py

tensor: Fix dimensions for fp16 in script

2024-08-12 15:22:07 -07:00

flash_attn.py

flash.py: Write V to file

2024-09-01 18:17:05 -07:00

generate_matrix.py

Generate S matrix, pull out FA stuff from basic script

2024-08-28 16:13:38 -07:00

main.cpp

tensor: Fix dimensions and makefile

2024-08-19 17:37:26 -07:00

Makefile

tensor: Fix dimensions and makefile

2024-08-19 17:37:26 -07:00

README.md

tensor: Update readme

2024-07-31 11:55:28 -07:00

test_data.h

bleh still not work

2024-03-27 00:26:04 -07:00

README.md

tensor

Unittest kernel for the tensor core.

Build

$ python3 generate_matrix.py
$ make

The generated ELF binary is run standalone; the argument and input matrix binary are hardcoded into the binary.