Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/FlashSampling/FlashSampling
/ persistent_matmul.py
File
persistent_matmul.py
src/fused_mm_sampling/persistent_matmul.py:None–None ·
view source on GitHub ↗
Source
from the content-addressed store, hash-verified
1
import
argparse
2
import
itertools
3
from
contextlib
import
contextmanager
4
from
typing
import
Optional
Callers
nothing calls this directly
Calls
5
is_cuda
Function · 0.85
is_hip
Function · 0.85
supports_tma
Function · 0.85
supports_ws
Function · 0.85
main
Function · 0.70
Tested by
no test coverage detected