MCPcopy Create free account
hub / github.com/FlashSampling/FlashSampling / supports_ws

Function supports_ws

src/fused_mm_sampling/persistent_matmul.py:47–48  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

45
46
47def supports_ws():
48 return is_cuda() and torch.cuda.get_device_capability()[0] >= 9
49
50
51def _matmul_launch_metadata(grid, kernel, args):

Callers 1

persistent_matmul.pyFile · 0.85

Calls 1

is_cudaFunction · 0.85

Tested by

no test coverage detected