File size: 1,697 Bytes
176b11a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
[
  "tests/test_patch.py::test_no_nan_output",
  "tests/test_patch.py::test_non_target_layer_no_pruning",
  "tests/test_patch.py::test_reset_n_vis",
  "tests/test_patch.py::test_wrapper_forward_shape",
  "tests/test_patch.py::test_wrapper_reduces_n_vis",
  "tests/test_rank_estimator.py::test_batched",
  "tests/test_rank_estimator.py::test_high_res",
  "tests/test_rank_estimator.py::test_known_low_rank",
  "tests/test_rank_estimator.py::test_prune_counts_valid",
  "tests/test_rank_estimator.py::test_single_matrix",
  "tests/test_scheduler.py::test_bucket_bounds",
  "tests/test_scheduler.py::test_bucket_idx",
  "tests/test_scheduler.py::test_make_scheduler",
  "tests/test_scheduler.py::test_snap_always_gte",
  "tests/test_scheduler.py::test_snap_in_buckets",
  "tests/test_scheduler.py::test_summary",
  "tests/test_sparse_attn.py::test_cpu_fallback",
  "tests/test_sparse_attn.py::test_high_compression",
  "tests/test_sparse_attn.py::test_matches_dense",
  "tests/test_sparse_attn.py::test_output_shape",
  "tests/test_token_scorer.py::test_prune_counts_bounds",
  "tests/test_token_scorer.py::test_rater_selection_shape",
  "tests/test_token_scorer.py::test_raters_above_mean",
  "tests/test_token_scorer.py::test_recycle_empty",
  "tests/test_token_scorer.py::test_recycle_output",
  "tests/test_token_scorer.py::test_score_shape",
  "tests/test_token_scorer.py::test_sparsevlm_score_no_nan",
  "tests/test_token_scorer.py::test_sparsevlm_score_shape",
  "tests/test_varlen.py::test_attention_mask",
  "tests/test_varlen.py::test_cu_seqlens",
  "tests/test_varlen.py::test_less_memory_than_padded",
  "tests/test_varlen.py::test_roundtrip",
  "tests/test_varlen.py::test_single_item"
]