Move harness back to tests

exo-lang · Apr 7, 2022 · 58917ff · 58917ff
1 parent 3a0cd0b
commit 58917ff
Show file tree

Hide file tree

Showing 11 changed files with 10 additions and 10 deletions.
diff --git a/tests/gemmini/conv/test_gemmini_conv.py b/tests/gemmini/conv/test_gemmini_conv.py
@@ -3,7 +3,7 @@
 import pytest
 
 from exo.platforms.gemmini import *
-from exo.platforms.harness_gemmini import GemmTestBuilder
+from ..harness_gemmini import GemmTestBuilder
 
 pytest.skip("skipping gemmini tests that are bitrotted",
             allow_module_level=True)

diff --git a/tests/gemmini/conv/test_gemmini_conv_ae.py b/tests/gemmini/conv/test_gemmini_conv_ae.py
@@ -1,7 +1,7 @@
 from __future__ import annotations
 import pytest
 from exo.platforms.gemmini import *
-from exo.platforms.harness_gemmini import GemmTestBuilder
+from ..harness_gemmini import GemmTestBuilder
 
 
 def conv_algorithm():

diff --git a/tests/gemmini/conv/test_gemmini_conv_max_pool.py b/tests/gemmini/conv/test_gemmini_conv_max_pool.py
@@ -3,7 +3,7 @@
 import pytest
 
 from exo.platforms.gemmini import *
-from exo.platforms.harness_gemmini import GemmTestBuilder
+from ..harness_gemmini import GemmTestBuilder
 
 pytest.skip("skipping gemmini tests that are bitrotted",
             allow_module_level=True)

diff --git a/tests/gemmini/conv/test_gemmini_conv_no_pad.py b/tests/gemmini/conv/test_gemmini_conv_no_pad.py
@@ -3,7 +3,7 @@
 import pytest
 
 from exo.platforms.gemmini import *
-from exo.platforms.harness_gemmini import GemmTestBuilder
+from ..harness_gemmini import GemmTestBuilder
 
 
 def conv_on_cpu():

diff --git a/tests/gemmini/conv/test_gemmini_conv_stride1.py b/tests/gemmini/conv/test_gemmini_conv_stride1.py
@@ -3,7 +3,7 @@
 import pytest
 
 from exo.platforms.gemmini import *
-from exo.platforms.harness_gemmini import GemmTestBuilder
+from ..harness_gemmini import GemmTestBuilder
 
 
 def conv_on_cpu():

diff --git a/tests/gemmini/conv/test_gemmini_conv_stride2.py b/tests/gemmini/conv/test_gemmini_conv_stride2.py
@@ -3,7 +3,7 @@
 import pytest
 
 from exo.platforms.gemmini import *
-from exo.platforms.harness_gemmini import GemmTestBuilder
+from ..harness_gemmini import GemmTestBuilder
 
 pytest.skip("skipping gemmini tests that are bitrotted",
             allow_module_level=True)

diff --git a/src/exo/platforms/harness_gemmini.py → tests/gemmini/harness_gemmini.py b/src/exo/platforms/harness_gemmini.py → tests/gemmini/harness_gemmini.py
diff --git a/tests/gemmini/matmul/test_gemmini_matmul_ae.py b/tests/gemmini/matmul/test_gemmini_matmul_ae.py
@@ -1,7 +1,7 @@
 from __future__ import annotations
 import pytest
 from exo.platforms.gemmini import *
-from exo.platforms.harness_gemmini import GemmTestBuilder
+from ..harness_gemmini import GemmTestBuilder
 
 
 def matmul_algorithm():

diff --git a/tests/gemmini/matmul/test_gemmini_matmul_paper.py b/tests/gemmini/matmul/test_gemmini_matmul_paper.py
@@ -3,7 +3,7 @@
 import pytest
 
 from exo.platforms.gemmini import *
-from exo.platforms.harness_gemmini import GemmTestBuilder
+from ..harness_gemmini import GemmTestBuilder
 
 
 def new_config_ld():

diff --git a/tests/gemmini/matmul/test_gemmini_resnet_matmul.py b/tests/gemmini/matmul/test_gemmini_resnet_matmul.py
@@ -3,7 +3,7 @@
 import pytest
 
 from exo.platforms.gemmini import *
-from exo.platforms.harness_gemmini import GemmTestBuilder
+from ..harness_gemmini import GemmTestBuilder
 
 def matmul_cpu():
     @proc

diff --git a/tests/gemmini/test_gemmini_instr.py b/tests/gemmini/test_gemmini_instr.py
@@ -3,7 +3,7 @@
 import pytest
 
 from exo.platforms.gemmini import *
-from exo.platforms.harness_gemmini import GemmTestBuilder
+from .harness_gemmini import GemmTestBuilder
 
 pytest.skip("skipping gemmini tests that are bitrotted",
             allow_module_level=True)