(backend: str, device: int)
| 78 | |
| 79 | |
| 80 | def sync_device(backend: str, device: int) -> None: |
| 81 | if backend == "cuda": |
| 82 | import torch |
| 83 | |
| 84 | torch.cuda.synchronize(device) |
| 85 | |
| 86 | |
| 87 | def audio_summary(audio: np.ndarray, sample_rate: int, wall_ms: float) -> dict[str, Any]: |