Minor fixes

Author: gramalingam

onnxscript/rewriter/ort_fusions/fuse_xformers_test.py

@@ -4,16 +4,14 @@
 
 import unittest
 
-from parameterized import parameterized
-
 import onnxscript.optimizer
+from onnxscript.rewriter.ort_fusions._core import fuse_xformers
 from onnxscript.rewriter.ort_fusions._smollm_1 import smollm_test_1
 from onnxscript.rewriter.ort_fusions._test_utils import assert_allclose, ort_run
-from onnxscript.rewriter.ort_fusions._core import fuse_xformers
 
-class TestTransformerFusion(unittest.TestCase):
 
-    def test_transformer_fusion(self):
+class TestFuseXformers(unittest.TestCase):
+    def test_fuse_xformers(self):
         test = smollm_test_1()
         model = test.get_onnx_model()
         onnxscript.optimizer.optimize(model)

onnxscript/rewriter/pattern.py

@@ -1566,7 +1566,7 @@ def copy_value(value: ir.Value | None) -> ir.Value | None:
             return None
         if value not in value_map:
             const_value = value.const_value
-            if isinstance(const_value, ir.Tensor):
+            if isinstance(const_value, (ir.Tensor, ir.TensorProtoTensor)):
                 # create a Constant node to represent the value
                 value_attr = ir.AttrTensor("value", const_value)
                 const_node = ir.Node("", "Constant", [], [value_attr])