Avoid accessing imag of non-complex tensors

torch/_inductor/decomposition.py

@@ -271,25 +271,27 @@ def angle(x):
 
 @register_decomposition([aten.add])
 def add(x, y, **kwargs):
-    x_is_complex = torch.is_tensor(x) and x.is_complex()
-    y_is_complex = torch.is_tensor(y) and y.is_complex()
-    if not x_is_complex and not y_is_complex:
+    x_is_complex_tensor = torch.is_tensor(x) and x.is_complex()
+    y_is_complex_tensor = torch.is_tensor(y) and y.is_complex()
+    if not x_is_complex_tensor and not y_is_complex_tensor:
         return NotImplemented
     r = y.real
-    i = y.imag
+    i = y.imag if y_is_complex_tensor else 0.0
     alpha = kwargs.get("alpha")
     if alpha is not None:
         r *= alpha
         i *= alpha
     r = x.real + r
-    i = x.imag + i
+    if x_is_complex_tensor:
+        i = x.imag + i
+    complex_type = x.dtype if x_is_complex_tensor else y.dtype
     return (
         torch.where(
             torch.arange(2, device=x.device, dtype=torch.uint8) == 0,
             r.unsqueeze(-1),
             i.unsqueeze(-1),
         )
-        .view(x.dtype)
+        .view(complex_type)
         .squeeze(-1)
     )