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

Function device_blas_name

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

Source from the content-addressed store, hash-verified

33
34
35def device_blas_name():
36 return "cuBLAS" if is_cuda() else "hipBLAS"
37
38
39def supports_tma():

Callers 3

device_blas_matmulFunction · 0.85
benchFunction · 0.85
validateFunction · 0.85

Calls 1

is_cudaFunction · 0.85

Tested by

no test coverage detected