| 23 | greedy_baseline = "greedy" |
| 24 | |
| 25 | |
| 26 | class LongNames: |
| 27 | fmms_triton = "FMMS (Triton)" |
| 28 | fmms_triton_p2p_no_overlap = "FMMS (Triton, P2P No Overlap)" |
| 29 | fmms_triton_ret_logits = "FMMS (Triton, Return Logits)" |
| 30 | fmms_greedy_triton = "FMMS Greedy (Triton)" |
| 31 | fmms_cuda = "FMMS (CUDA)" |
| 32 | fmms_topk = "FMMS Top-k (Triton)" |
| 33 | fmms_helion = "FMMS (Helion)" |
| 34 | multinomial_sampling_compiled = "Multinomial Sampling (Compiled)" |
| 35 | multinomial_sampling_eager = "Multinomial Sampling (Eager)" |
| 36 | sequential_compiled = "Sequential PyTorch Compiled" |
| 37 | naive_tl_matmul = "Naive Triton Matmul" |
| 38 | jl_compiled = "JL Compiled" |
| 39 | pt_qitra = "Qitra" |
| 40 | flashinfer_top_k_top_p_sampling_from_logits = "flashinfer:top_k_top_p_sampling_from_logits" |
| 41 | flashinfer_sampling_from_logits = "flashinfer:sampling_from_logits" |
| 42 | greedy_baseline = "Greedy Baseline" |
| 43 |
nothing calls this directly
no outgoing calls
no test coverage detected