(self, warmup_runs: int = 5, benchmark_runs: int = 100, tokenizer_type: str = "llama")
| 66 | """Comprehensive performance benchmark suite.""" |
| 67 | |
| 68 | def __init__(self, warmup_runs: int = 5, benchmark_runs: int = 100, tokenizer_type: str = "llama"): |
| 69 | self.src_dir = Path(__file__).parent.parent / "src" |
| 70 | self.test_dir = Path(__file__).parent |
| 71 | self.warmup_runs = warmup_runs |
| 72 | self.benchmark_runs = benchmark_runs |
| 73 | self.tokenizer_type = tokenizer_type.lower() |
| 74 | self.results: List[BenchmarkResult] = [] |
| 75 | |
| 76 | # Validate tokenizer type |
| 77 | if self.tokenizer_type not in ["llama", "mistral"]: |
| 78 | raise ValueError(f"Invalid tokenizer type: {tokenizer_type}. Must be 'llama' or 'mistral'") |
| 79 | |
| 80 | def load_llama_config(self) -> Tuple[str, List[Dict[str, Any]], Dict[str, int]]: |
| 81 | """Load Llama 4 configuration from the codebase.""" |
nothing calls this directly
no outgoing calls
no test coverage detected