Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/FlashSampling/FlashSampling
/ functions
Functions
315 in github.com/FlashSampling/FlashSampling
⨍
Functions
315
◇
Types & classes
25
↳
Endpoints
9
Function
test_fused_triton_return_logits
Verify that the fused kernel returns logits matching a PyTorch matmul reference.
tests/test_core.py:199
Function
test_gemv
()
tests/test_gemv.py:6
Function
test_greedy_sampling
Verify that greedy_sampling=True returns the argmax token for each sequence.
tests/test_core.py:220
Function
test_greedy_tp2
()
tests/test_core.py:98
Function
test_jl_sampling_aproximate_correctness
()
tests/test_core.py:23
Function
test_matmul
()
tests/test_matmul.py:6
Function
test_sampling_distribution
(provider, vocab_size, n_hidden_states)
tests/test_core.py:74
Function
test_sampling_distribution_tp2
()
tests/test_core.py:86
Function
test_speed_test_smoke
()
tests/test_core.py:239
Function
test_top_k_top_p
Verify that top-k and top-p filtering restricts samples to the expected tokens.
tests/test_core.py:105
Function
test_top_k_top_p_large_vocab
Test top-k + top-p at V=151936 with real Qwen3-0.6b weights. At large vocab sizes, tl.dot and cuBLAS produce different logits due to bf16 acc
tests/test_core.py:142
Function
tl_gemv_kernel
( x_ptr, y_ptr, out_ptr, M: tl.constexpr, # noqa: N803 B: tl.constexpr, # noqa: N803
src/fused_mm_sampling/tl_gemv.py:46
Function
torch_matmul
(a, b)
src/fused_mm_sampling/persistent_matmul.py:686
Function
torch_matmul_nontransposed
(a, b)
src/fused_mm_sampling/persistent_matmul.py:699
Function
verify_correctness_tp2
()
src/fused_mm_sampling/testing.py:162
← previous
301–315 of 315, ranked by callers