diff --git a/tests/gemmini/conv/test_gemmini_conv.py b/tests/gemmini/conv/test_gemmini_conv.py index 0bca4a252..08f0ae5ef 100644 --- 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 index 608fe9f67..05fa39d9c 100644 --- 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 index bbc8a89ea..4cbc87885 100644 --- 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 index 016e1ca37..08fdc04bc 100644 --- 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 index f3c9ab558..d6e540c65 100644 --- 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 index 917a6583e..08be3377b 100644 --- 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 b/tests/gemmini/harness_gemmini.py similarity index 100% rename from src/exo/platforms/harness_gemmini.py rename to 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 index d1869d0d3..3321e6841 100644 --- 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 index 7e5cb8641..5f3fd0000 100644 --- 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 index a45c845ad..b1a8f8f32 100644 --- 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 index 5b8a3a919..4e2e2cc48 100644 --- 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)