nki-samples/test/unit at main · FPSG-UIUC/nki-samples

Name		Name	Last commit message	Last commit date
parent directory ..
README.md		README.md
conftest.py		conftest.py
test_SD_attention_small_head.py		test_SD_attention_small_head.py
test_allocated_SD_attention_small_head.py		test_allocated_SD_attention_small_head.py
test_flash_attn_bwd.py		test_flash_attn_bwd.py
test_flash_attn_fwd.py		test_flash_attn_fwd.py
test_neuron_profile.py		test_neuron_profile.py
test_resize_nearest.py		test_resize_nearest.py
test_rmsnorm_qkv.py		test_rmsnorm_qkv.py
test_select_and_scatter.py		test_select_and_scatter.py

README.md

Tests under this folder are unit tests for the kernels in src/nki_samples.

To execute the tests, we need to include src/nki_samples in the PYTHONPATH.

For example,

PYTHONPATH=$PYTHONPATH:/home/ubuntu/nki-samples/src/ pytest test_flash_attn_fwd.py