Library name to import is `flash_attn_interface`, which matches the test.
Attach a file by drag & drop or click to upload
