| 32 | |
| 33 | |
| 34 | class AttentionImplementations(Enum): |
| 35 | ORIGINAL = "ORIGINAL" |
| 36 | SPLIT_EINSUM = "SPLIT_EINSUM" |
| 37 | SPLIT_EINSUM_V2 = "SPLIT_EINSUM_V2" |
| 38 | |
| 39 | |
| 40 | ATTENTION_IMPLEMENTATION_IN_EFFECT = AttentionImplementations.SPLIT_EINSUM |
nothing calls this directly
no outgoing calls
no test coverage detected