[OpInfo][BE] Better inputs for addmm (#160234)

malfet · pytorchmergebot · commit df55ec7d4b35 · 2025-08-10T01:26:48.000Z
Right now alpha and betha are both less than zero, which makes them useless for all addmm samples for interal types Pull Request resolved: #160234 Approved by: https://github.com/Skylion007 ghstack dependencies: #160228
diff --git a/torch/testing/_internal/common_methods_invocations.py b/torch/testing/_internal/common_methods_invocations.py
@@ -1161,8 +1161,8 @@ def make_arg_conj(size):
 
 
 def sample_inputs_addmm(op_info, device, dtype, requires_grad, **kwargs):
-    alpha_val = kwargs.get('alpha', 2 + 3j if dtype.is_complex else 0.6)
-    beta_val = kwargs.get('beta', 1 + 2j if dtype.is_complex else 0.2)
+    alpha_val = kwargs.get('alpha', 2 + 3j if dtype.is_complex else 0.6 if dtype.is_floating_point else 2)
+    beta_val = kwargs.get('beta', 1 + 2j if dtype.is_complex else 0.2 if dtype.is_floating_point else 3)
     tests_list = [
         ((2, 3), (2, 2), (2, 3), False),
         ((3, 3), (3, 3), (3, 3), False),
