Refactor test models for ort_fusions (#2237)

Moving test models used for ort fusions to models folder for to avoid
bloating of root folder when more test models (for example,
whisper-decoder and whisper-encoder) are added to test models

Author: shubhambhokare1

onnxscript/rewriter/ort_fusions/cos_sin_cache_test.py

@@ -7,14 +7,14 @@
 from parameterized import parameterized
 
 import onnxscript.optimizer
-from onnxscript.rewriter.ort_fusions._rotary_embedding_models import (
+from onnxscript.rewriter.ort_fusions._test_utils import assert_allclose, ort_run
+from onnxscript.rewriter.ort_fusions.cos_sin_cache import fuse_cos_sin_cache
+from onnxscript.rewriter.ort_fusions.models._rotary_embedding_models import (
     partial_rotary_test_case,
     test_case_1,
     test_case_2,
 )
-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.cos_sin_cache import fuse_cos_sin_cache
+from onnxscript.rewriter.ort_fusions.models._smollm_1 import smollm_test_1
 from onnxscript.rewriter.ort_fusions.rotary_embedding import (
     fuse_partial_rotary_embedding,
     fuse_rotary_embedding,

onnxscript/rewriter/ort_fusions/fuse_xformers_test.py

@@ -6,8 +6,8 @@
 
 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.models._smollm_1 import smollm_test_1
 
 
 class TestFuseXformers(unittest.TestCase):

onnxscript/rewriter/ort_fusions/mha_test.py

@@ -8,8 +8,8 @@
 
 import onnxscript.optimizer
 import onnxscript.rewriter.ort_fusions._core as xformers
-from onnxscript.rewriter.ort_fusions._smollm_2 import smollm_test_2
 from onnxscript.rewriter.ort_fusions._test_utils import ORT_VERSION, assert_allclose, ort_run
+from onnxscript.rewriter.ort_fusions.models._smollm_2 import smollm_test_2
 
 
 class TestMultiHeadAttention(unittest.TestCase):

onnxscript/rewriter/ort_fusions/_rotary_embedding_models.py renamed to onnxscript/rewriter/ort_fusions/models/_rotary_embedding_models.py

@@ -5,8 +5,8 @@
 import unittest
 
 import onnxscript.optimizer
-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.models._smollm_1 import smollm_test_1
 from onnxscript.rewriter.ort_fusions.rms_normalization import fuse_rms_normalization
 
 

onnxscript/rewriter/ort_fusions/_smollm_1.py renamed to onnxscript/rewriter/ort_fusions/models/_smollm_1.py

@@ -7,8 +7,8 @@
 from parameterized import parameterized
 
 import onnxscript.optimizer
-from onnxscript.rewriter.ort_fusions._rotary_embedding_models import test_case_1
-from onnxscript.rewriter.ort_fusions._smollm_1 import smollm_test_1
+from onnxscript.rewriter.ort_fusions.models._rotary_embedding_models import test_case_1
+from onnxscript.rewriter.ort_fusions.models._smollm_1 import smollm_test_1
 from onnxscript.rewriter.ort_fusions.rotary_embedding import fuse_rotary_embedding
 
 

onnxscript/rewriter/ort_fusions/_smollm_2.py renamed to onnxscript/rewriter/ort_fusions/models/_smollm_2.py

@@ -5,8 +5,8 @@
 import unittest
 
 import onnxscript.optimizer
-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.models._smollm_1 import smollm_test_1
 from onnxscript.rewriter.ort_fusions.rms_normalization import fuse_rms_normalization
 from onnxscript.rewriter.ort_fusions.skip_normalization import fuse_skip_rms_normalization