Path	Lines of Code
experimental/ragged_inference/test_utils.py	62
experimental/tests/test_seq_kv_cache.py	179
experimental/tests/test_triton_v2_matmul.py	19
experimental/tests/test_triton_v2_qk_dotprod.py	81
experimental/tests/test_triton_v2_ragged_qk_dotprod.py	134
stubs/torch_stub_tests.py	1117
tests/__init__.py	1
tests/test_attention_mask.py	36
tests/test_attention_patterns.py	136
tests/test_attention_utils.py	28
tests/test_attentions.py	254
tests/test_block_factory.py	185
tests/test_compositional_attention.py	83
tests/test_core_attention.py	67
tests/test_custom_ops.py	337
tests/test_embedding.py	28
tests/test_favor.py	111
tests/test_feedforward.py	32
tests/test_global_attention.py	22
tests/test_hydra_helper.py	9
tests/test_model_factory.py	119
tests/test_nystrom_attention.py	86
tests/test_ortho_attention.py	64
tests/test_pytorch_transformer_parity.py	167
tests/test_reversible.py	168
tests/test_rotary_embeddings.py	31
tests/test_sparse_csr.py	93
tests/test_sparsecs.py	62
tests/test_timm_wrap.py	45
tests/test_triton_basics.py	85
tests/test_triton_blocksparse.py	213
tests/test_triton_dropout.py	129
tests/test_triton_fused_linear.py	101
tests/test_triton_layernorm.py	75
tests/test_triton_softmax.py	107