in llm_perf/benchmark_runners/cpu/update_llm_perf_cpu_pytorch.py [0:0]
def __init__(self):
super().__init__(backend="pytorch", device="cpu")
self.attention_configs = self._get_attention_configs()
assert (
self.subset is not None
), "SUBSET environment variable must be set for benchmarking"
self.weights_configs = self._get_weights_configs(self.subset)