hide warnings

Author: xadupre

_unittests/ut_reference/test_torch_onnx_evaluator.py

@@ -1377,6 +1377,7 @@ def test_tile(self):
             torch.tensor([2, 2], dtype=torch.int64),
         )
 
+    @ignore_warnings(UserWarning)
     def test_custom_kernels(self):
         class LayerNormalizationOrt(OpRunKernel):
             "LayerNormalization"

_unittests/ut_torch_export_patches/test_patch_module.py

@@ -5,7 +5,13 @@
 import numpy as np
 from scipy.spatial.distance import cdist
 import torch
-from onnx_diagnostic.ext_test_case import ExtTestCase, hide_stdout, has_torch, requires_torch
+from onnx_diagnostic.ext_test_case import (
+    ExtTestCase,
+    hide_stdout,
+    has_torch,
+    requires_torch,
+    ignore_warnings,
+)
 from onnx_diagnostic.torch_export_patches import torch_export_patches, torch_export_rewrite
 from onnx_diagnostic.torch_export_patches.patch_module import (
     transform_method,
@@ -370,6 +376,7 @@ def forward(self, x, y):
         self.assertEqualAny(expected_0, ep.module()(x, -y))
         self.assertEqualAny(expected_1, ep.module()(-x, -y))
 
+    @ignore_warnings(UserWarning)
     def test_rewrite_test_in_forward_none(self):
 
         class Model(torch.nn.Module):

onnx_diagnostic/torch_export_patches/patches/patch_transformers.py

@@ -1032,7 +1032,8 @@ def patched_modeling_marian_eager_attention_forward(
 
 
 class common_RotaryEmbedding(torch.nn.Module):
-    @torch.no_grad()
+    # This may cause some issues.
+    # @torch.no_grad()
     @patched_dynamic_rope_update
     def forward(self, x, position_ids):
         inv_freq_expanded = (