Home
last modified time | relevance | path

Searched refs:autoheuristic_use (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/test/inductor/
Dtest_autoheuristic.py117 @inductor_config.patch(autoheuristic_use="pad_mm")
124 @inductor_config.patch(autoheuristic_use="pad_mm")
145 autoheuristic_use="",
159 @inductor_config.patch(autoheuristic_use="mixed_mm")
Dtest_pattern_matcher.py306 autoheuristic_use="",
366 autoheuristic_use="",
/external/pytorch/torchgen/_autoheuristic/
Dbenchmark_runner.py59 torch._inductor.config.autoheuristic_use = self.name
62 torch._inductor.config.autoheuristic_use = ""
/external/pytorch/torch/_inductor/
Dconfig.py364 autoheuristic_use = os.environ.get("TORCHINDUCTOR_AUTOHEURISTIC_USE", "mixed_mm") variable
376 return name in torch._inductor.config.autoheuristic_use.split(",")