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

Function _as_torch_stream

src/fused_mm_sampling/bench/speed_test.py:203–204  ·  view source on GitHub ↗
(cs: "nvbench.CudaStream")

Source from the content-addressed store, hash-verified

201
202
203def _as_torch_stream(cs: "nvbench.CudaStream") -> torch.cuda.ExternalStream:
204 return torch.cuda.ExternalStream(cs.addressof())
205
206
207def run_cupti(args: Args) -> None:

Callers 1

launcherFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected