From f8a2542e12bd88d830d7a92be5ac2257a52d2829 Mon Sep 17 00:00:00 2001
From: Pablo Monteagudo Lago <pablo.monteagudolago@amd.com>
Date: Fri, 6 Dec 2024 12:24:27 +0000
Subject: [PATCH] Remove not used function

---
 tests/brevitas_examples/test_llm.py | 9 ---------
 1 file changed, 9 deletions(-)

diff --git a/tests/brevitas_examples/test_llm.py b/tests/brevitas_examples/test_llm.py
index ac9bd6ec2..dc5d42a03 100644
--- a/tests/brevitas_examples/test_llm.py
+++ b/tests/brevitas_examples/test_llm.py
@@ -291,15 +291,6 @@ def test_small_models_acc_pt_ge_2_4(caplog, acc_args_and_acc_pt_ge_2_4):
     assert allveryclose(exp_quant_ppl, quant_ppl), f"Expected quant PPL {exp_quant_ppl}, measured PPL {quant_ppl}"
 
 
-def llama_module_type_count(num_hidden_layers: int):
-    return {
-        "<class 'torch.nn.modules.linear.Linear'>": 1 + num_hidden_layers *
-                                                    (4 + 3),  # lm_head + (k/q/v/o_proj + MLP)
-        "<class 'torch.nn.modules.normalization.RMSNorm'>":
-            1 + num_hidden_layers * 2,  # input + post_attention
-    }
-
-
 @pytest_cases.fixture(
     ids=[
         "mistral-int8",