Searched defs:benchmark_kernel (Results 1 – 3 of 3) sorted by relevance
94 def generate_kernel_code_from_nodes(self, nodes, benchmark_kernel=False): argument
1770 def generate_kernel_code_from_nodes(self, nodes, benchmark_kernel=False): argument
457 benchmark_kernel = os.environ.get("TORCHINDUCTOR_BENCHMARK_KERNEL", "0") == "1" variable