diff --git a/tests/scripts/run_python_model_tests.sh b/tests/scripts/run_python_model_tests.sh index 71a286e125b..e3e6674a633 100755 --- a/tests/scripts/run_python_model_tests.sh +++ b/tests/scripts/run_python_model_tests.sh @@ -35,7 +35,7 @@ run_python_model_tests_wormhole_b0() { # higher sequence lengths and different formats trigger memory issues pytest models/demos/falcon7b_common/tests/unit_tests/test_falcon_matmuls_and_bmms_with_mixed_precision.py -k "seq_len_128 and in0_BFLOAT16-in1_BFLOAT8_B-out_BFLOAT16-weights_DRAM" pytest tests/ttnn/integration_tests/resnet/test_ttnn_functional_resnet50_new.py -k "pretrained_weight_false" - pytest models/demo/yolov4/demo/demo.py -k "pretrained_weight_false" + pytest models/demos/yolov4/demo/demo.py -k "pretrained_weight_false" # Unet Shallow WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml pytest -svv models/experimental/functional_unet/tests/test_unet_model.py diff --git a/tests/ttnn/integration_tests/yolov4/test_ttnn_downsample1.py b/tests/ttnn/integration_tests/yolov4/test_ttnn_downsample1.py index f5e7f834b85..b1dfa163326 100644 --- a/tests/ttnn/integration_tests/yolov4/test_ttnn_downsample1.py +++ b/tests/ttnn/integration_tests/yolov4/test_ttnn_downsample1.py @@ -5,8 +5,8 @@ import ttnn import torch from models.utility_functions import skip_for_grayskull -from models.experimental.yolov4.reference.downsample1 import DownSample1 -from models.experimental.yolov4.ttnn.downsample1 import Down1 +from models.demos.yolov4.reference.downsample1 import DownSample1 +from models.demos.yolov4.ttnn.downsample1 import Down1 from tests.ttnn.utils_for_testing import assert_with_pcc import pytest import time diff --git a/tests/ttnn/integration_tests/yolov4/test_ttnn_downsample2.py b/tests/ttnn/integration_tests/yolov4/test_ttnn_downsample2.py index e8ad8264d01..715f9ddf259 100644 --- a/tests/ttnn/integration_tests/yolov4/test_ttnn_downsample2.py +++ b/tests/ttnn/integration_tests/yolov4/test_ttnn_downsample2.py @@ -8,8 +8,8 @@ import pytest import time from models.utility_functions import skip_for_grayskull -from models.experimental.yolov4.reference.downsample2 import DownSample2 -from models.experimental.yolov4.ttnn.downsample2 import Down2 +from models.demos.yolov4.reference.downsample2 import DownSample2 +from models.demos.yolov4.ttnn.downsample2 import Down2 from loguru import logger import os diff --git a/tests/ttnn/integration_tests/yolov4/test_ttnn_downsample3.py b/tests/ttnn/integration_tests/yolov4/test_ttnn_downsample3.py index cf5b78c9673..88225429b5a 100644 --- a/tests/ttnn/integration_tests/yolov4/test_ttnn_downsample3.py +++ b/tests/ttnn/integration_tests/yolov4/test_ttnn_downsample3.py @@ -8,8 +8,8 @@ import pytest import time from models.utility_functions import skip_for_grayskull -from models.experimental.yolov4.reference.downsample3 import DownSample3 -from models.experimental.yolov4.ttnn.downsample3 import Down3 +from models.demos.yolov4.reference.downsample3 import DownSample3 +from models.demos.yolov4.ttnn.downsample3 import Down3 from loguru import logger import os diff --git a/tests/ttnn/integration_tests/yolov4/test_ttnn_downsample4.py b/tests/ttnn/integration_tests/yolov4/test_ttnn_downsample4.py index 30d5da1cc82..825f08482fe 100644 --- a/tests/ttnn/integration_tests/yolov4/test_ttnn_downsample4.py +++ b/tests/ttnn/integration_tests/yolov4/test_ttnn_downsample4.py @@ -8,8 +8,8 @@ import pytest import time from models.utility_functions import skip_for_grayskull -from models.experimental.yolov4.reference.downsample4 import DownSample4 -from models.experimental.yolov4.ttnn.downsample4 import Down4 +from models.demos.yolov4.reference.downsample4 import DownSample4 +from models.demos.yolov4.ttnn.downsample4 import Down4 from loguru import logger import os diff --git a/tests/ttnn/integration_tests/yolov4/test_ttnn_downsample5.py b/tests/ttnn/integration_tests/yolov4/test_ttnn_downsample5.py index eb5d8f10cdf..df0c1fb4709 100644 --- a/tests/ttnn/integration_tests/yolov4/test_ttnn_downsample5.py +++ b/tests/ttnn/integration_tests/yolov4/test_ttnn_downsample5.py @@ -8,8 +8,8 @@ import pytest import time from models.utility_functions import skip_for_grayskull -from models.experimental.yolov4.reference.downsample5 import DownSample5 -from models.experimental.yolov4.ttnn.downsample5 import Down5 +from models.demos.yolov4.reference.downsample5 import DownSample5 +from models.demos.yolov4.ttnn.downsample5 import Down5 from loguru import logger import os diff --git a/tests/ttnn/integration_tests/yolov4/test_ttnn_head.py b/tests/ttnn/integration_tests/yolov4/test_ttnn_head.py index 98e8beace11..5d1f31da9c2 100644 --- a/tests/ttnn/integration_tests/yolov4/test_ttnn_head.py +++ b/tests/ttnn/integration_tests/yolov4/test_ttnn_head.py @@ -4,11 +4,11 @@ import torch import ttnn -from models.experimental.yolov4.reference.head import Head +from models.demos.yolov4.reference.head import Head from tests.ttnn.utils_for_testing import assert_with_pcc import pytest import time -from models.experimental.yolov4.ttnn.head import TtHead +from models.demos.yolov4.ttnn.head import TtHead from loguru import logger import os diff --git a/tests/ttnn/integration_tests/yolov4/test_ttnn_neck.py b/tests/ttnn/integration_tests/yolov4/test_ttnn_neck.py index 74268531233..01e9fe4bf96 100644 --- a/tests/ttnn/integration_tests/yolov4/test_ttnn_neck.py +++ b/tests/ttnn/integration_tests/yolov4/test_ttnn_neck.py @@ -4,8 +4,8 @@ import torch import ttnn -from models.experimental.yolov4.ttnn.neck import TtNeck -from models.experimental.yolov4.reference.neck import Neck +from models.demos.yolov4.ttnn.neck import TtNeck +from models.demos.yolov4.reference.neck import Neck from tests.ttnn.utils_for_testing import assert_with_pcc import pytest import time diff --git a/tests/ttnn/integration_tests/yolov4/test_ttnn_yolov4.py b/tests/ttnn/integration_tests/yolov4/test_ttnn_yolov4.py index 7e4a4e4933e..185ba8ed6b6 100644 --- a/tests/ttnn/integration_tests/yolov4/test_ttnn_yolov4.py +++ b/tests/ttnn/integration_tests/yolov4/test_ttnn_yolov4.py @@ -5,9 +5,9 @@ import torch import ttnn from models.utility_functions import skip_for_grayskull -from models.experimental.yolov4.reference.yolov4 import Yolov4 +from models.demos.yolov4.reference.yolov4 import Yolov4 from tests.ttnn.utils_for_testing import assert_with_pcc -from models.experimental.yolov4.ttnn.yolov4 import TtYOLOv4 +from models.demos.yolov4.ttnn.yolov4 import TtYOLOv4 import pytest import os