From 75d3bbaaa23a5b3febb0264df0742b6b6e4c916e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Carlos=20Mochol=C3=AD?= Date: Wed, 13 Dec 2023 18:44:38 +0000 Subject: [PATCH] Fix cudagraph check message (#115664) This error message is printed when CUDAGraph trees are used with multiple device indices. However, the message seems to say the opposite. Pull Request resolved: https://github.com/pytorch/pytorch/pull/115664 Approved by: https://github.com/soulitzer --- torch/_inductor/compile_fx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch/_inductor/compile_fx.py b/torch/_inductor/compile_fx.py index bc38ca1b9a2bf2..481c4b679cd20a 100644 --- a/torch/_inductor/compile_fx.py +++ b/torch/_inductor/compile_fx.py @@ -369,7 +369,7 @@ def compile_fx_inner( len(compiled_graph.device_idxs) == 1 or not config.triton.cudagraph_trees ), - "multiple device indices without cudagraph_trees", + "multiple device indices with cudagraph_trees", ), ] cudagraph_fail_reasons = [s for b, s in cudagraph_tests if not b]