Skip to content

Commit 24e9a7d

Browse files
committed
Merge branch 'perf' of https://github.com/roboflow/rf-Cutie into perf
2 parents 95f98af + 842c7ef commit 24e9a7d

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

‎scripts/bench_vram_mps.py‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,8 @@ def _run_once(mk: torch.Tensor, ms: torch.Tensor, qk: torch.Tensor, qe: torch.Te
3636

3737

3838
def bench(*, batch: int, ck: int, num_memory_frames: int, hw: int, iters: int) -> None:
39+
if iters < 1:
40+
raise ValueError('iters must be >= 1')
3941
if not torch.backends.mps.is_available():
4042
print('MPS not available on this machine — nothing to bench.')
4143
return

0 commit comments

Comments
 (0)