Print output type if assert fires (#110534)

ezyang · pytorchmergebot · commit f1d81134ef59 · 2023-10-05T00:59:17.000Z
Signed-off-by: Edward Z. Yang <ezyang@meta.com> Pull Request resolved: #110534 Approved by: https://github.com/albanD
diff --git a/torch/_inductor/ir.py b/torch/_inductor/ir.py
@@ -3985,7 +3985,9 @@ def generate_output(output, indices):
             elif isinstance(output, int):
                 return output
             else:
-                assert output is None, "FallbackKernel output type is not supported"
+                assert (
+                    output is None
+                ), f"FallbackKernel output type {type(output)} is not supported"
                 return None
 
         outputs = generate_output(example_output, [])
