Antoni BaumandGitHub cf5cb1e33e Allocate more shared memory to attention kernel (#1154)

2023-09-26 22:27:13 -07:00

..

conftest.py

Clean up kernel unit tests (#938)

2023-09-05 16:57:38 -07:00

test_activation.py

Clean up kernel unit tests (#938)

2023-09-05 16:57:38 -07:00

test_attention.py

Allocate more shared memory to attention kernel (#1154)

2023-09-26 22:27:13 -07:00

test_cache.py

Clean up kernel unit tests (#938)

2023-09-05 16:57:38 -07:00

test_layernorm.py

Clean up kernel unit tests (#938)

2023-09-05 16:57:38 -07:00

test_pos_encoding.py

Use FP32 in RoPE initialization (#1004)

2023-09-11 00:26:35 -07:00