MCPcopy
hub / github.com/z-lab/dflash / _cuda_time

Function _cuda_time

dflash/model.py:57–59  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

55
56
57def _cuda_time() -> float:
58 torch.cuda.synchronize()
59 return time.perf_counter()
60
61
62@torch.inference_mode()

Callers 1

dflash_generateFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected