Wing Lian 394a65f11f add unit tests for cum seq lens, add ability to build cu_seq_lens from positional ids, fix prompt test

2023-08-07 09:38:04 -04:00

..

test_llama_attn_hijack_flash.py

add unit tests for cum seq lens, add ability to build cu_seq_lens from positional ids, fix prompt test

2023-08-07 09:38:04 -04:00