Searched defs:autotune_multi_device (Results 1 – 2 of 2) sorted by relevance
153 def test_max_autotune_mm_plus_mm(self, autotune_in_subproc, autotune_multi_device): argument
347 autotune_multi_device = os.environ.get("TORCHINDUCTOR_AUTOTUNE_MULTI_DEVICE") == "1" variable