Skip to content

Commit

Permalink
Fix cudagraph check message (pytorch#115664)
Browse files Browse the repository at this point in the history
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: pytorch#115664
Approved by: https://github.com/soulitzer
  • Loading branch information
carmocca authored and pytorchmergebot committed Dec 13, 2023
1 parent 42390a0 commit 75d3bba
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torch/_inductor/compile_fx.py
Original file line number Diff line number Diff line change
Expand Up @@ -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]
Expand Down

0 comments on commit 75d3bba

Please sign in to comment.