#0: testing

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

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

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
 

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
 

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
 

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
 

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
 

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

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