File size: 308 Bytes
7c85c7e
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
"""CPU-only wrapper for the pinned public JevBench speed protocol."""
from pathlib import Path
import sys

sys.path.insert(0, str(Path(__file__).resolve().parents[1] / "benchmarks"))
from jevbench_public import main

if __name__ == "__main__":
    main([*sys.argv[1:], "--device", "cpu", "--mode", "speed"])