From 56343b28f69c08ac0670bcaa95874e5475bd74aa Mon Sep 17 00:00:00 2001 From: Shwetank Singh Date: Sat, 13 Jul 2024 19:20:41 +0000 Subject: [PATCH] #0: testing --- .../test_ttnn_functional_resnet50_new.py | 21 +++++++++++++++++-- 1 file changed, 19 insertions(+), 2 deletions(-) diff --git a/tests/ttnn/integration_tests/resnet/test_ttnn_functional_resnet50_new.py b/tests/ttnn/integration_tests/resnet/test_ttnn_functional_resnet50_new.py index fd3d1b86cce..36eb749c010 100644 --- a/tests/ttnn/integration_tests/resnet/test_ttnn_functional_resnet50_new.py +++ b/tests/ttnn/integration_tests/resnet/test_ttnn_functional_resnet50_new.py @@ -133,7 +133,15 @@ class ResNet50TestInfra: def __init__( - self, device, batch_size, act_dtype, weight_dtype, math_fidelity, use_pretrained_weight, dealloc_input, final_output_mem_config + self, + device, + batch_size, + act_dtype, + weight_dtype, + math_fidelity, + use_pretrained_weight, + dealloc_input, + final_output_mem_config, ): super().__init__() torch.manual_seed(0) @@ -234,7 +242,14 @@ def create_test_infra( final_output_mem_config=ttnn.L1_MEMORY_CONFIG, ): return ResNet50TestInfra( - device, batch_size, act_dtype, weight_dtype, math_fidelity, use_pretrained_weight, dealloc_input, final_output_mem_config + device, + batch_size, + act_dtype, + weight_dtype, + math_fidelity, + use_pretrained_weight, + dealloc_input, + final_output_mem_config, ) @@ -269,6 +284,8 @@ def test_resnet_50( pytest.skip("Skipping batch size 8 due to memory config issue") if is_wormhole_b0() and batch_size == 20: pytest.skip("Skipping batch size 20 for Wormhole B0 due to fitting issue") + if (device.compute_with_storage_grid_size().x, device.compute_with_storage_grid_size().y) == (8, 7): + pytest.skip("Test is not supported on n300 (8,7) grid") test_infra = create_test_infra(device, batch_size, act_dtype, weight_dtype, math_fidelity, use_pretrained_weight) enable_memory_reports()