Skip to content

Checking we use fused kernels to compute scaled masked softmax on prefix lm#213

Open
thomasw21 wants to merge 6 commits intomainfrom thomas/improve_test_to_test_custom_kernel