Files

Hansung Kim 2f7fb372f1 Fix range for

i'm a python noob

2024-08-19 21:19:36 -07:00

check_correctness.py

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

flash_attn.py

Fix range for

2024-08-19 21:19:36 -07:00

generate_matrix.py

2024-08-19 21:16:37 -07:00

main.cpp

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

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

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.