Format

Signed-off-by: Justin Chu <[email protected]>

Author: justinchuby

tests/function_libs/torch_lib/extra_opinfo.py

@@ -48,10 +48,10 @@ def sample_inputs_bilinear(op_info, device, dtype, requires_grad, **kwargs):
 
     # Test cases: (batch_size, in1_features, in2_features, out_features)
     cases = [
-        (2, 3, 4, 5),      # Basic case
-        (1, 2, 2, 1),      # Minimal case
-        (3, 5, 7, 4),      # Different dimensions
-        (2, 1, 1, 3),      # Single input features
+        (2, 3, 4, 5),  # Basic case
+        (1, 2, 2, 1),  # Minimal case
+        (3, 5, 7, 4),  # Different dimensions
+        (2, 1, 1, 3),  # Single input features
     ]
 
     for batch_size, in1_features, in2_features, out_features in cases:

tests/function_libs/torch_lib/ops_test_data.py

@@ -657,7 +657,9 @@ def _where_input_wrangler(
     ),
     TorchLibOpInfo("baddbmm", core_ops.aten_baddbmm, tolerance={torch.float16: (1e-3, 1e-2)}),
     TorchLibOpInfo("bernoulli", core_ops.aten_bernoulli, nondeterministic=True),
-    TorchLibOpInfo("bilinear", core_ops.aten_bilinear, tolerance={torch.float32: (1e-4, 1e-4)}),
+    TorchLibOpInfo(
+        "bilinear", core_ops.aten_bilinear, tolerance={torch.float32: (2e-5, 2e-5)}
+    ),
     TorchLibOpInfo(
         # This string is a unique ID. In extra_opinfo.py, we
         # also define test data for this ID with