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

Function is_cuda

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

Source from the content-addressed store, hash-verified

11
12
13def is_cuda():
14 return triton.runtime.driver.active.get_current_target().backend == "cuda"
15
16
17def is_hip():

Callers 5

persistent_matmul.pyFile · 0.85
device_blas_nameFunction · 0.85
supports_tmaFunction · 0.85
supports_wsFunction · 0.85
mainFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected