fix test nondeterminism

blaine-rister · blaine-rister · commit 9c5d20284ac7 · 2024-12-06T13:12:00.000-08:00
diff --git a/test/inductor/test_op_dtype_prop.py b/test/inductor/test_op_dtype_prop.py
@@ -168,8 +168,9 @@ def test_dtype_aware_codegen(self, op_name: str, load_upcast_to_fp32, input_dtyp
         triton_op_name = override if override is not None else torch_op_name
 
         # Get the number of args for the op.
+        # Take the minimum over all signatures to isolate required args.
         signatures = get_signature_for_torch_op(op)
-        num_args = len(signatures[0].parameters)
+        num_args = min(len(signature.parameters) for signature in signatures)
 
         # Test codegen and check for casts.
         inps = (torch.rand((32, 32), device=GPU_TYPE, dtype=input_dtype),) * num_args
