[Kernel] Add punica dimensions for Granite 13b (#5559)

Signed-off-by: Joe Runde <Joseph.Runde@ibm.com>
2024-06-17 21:54:11 -06:00
parent daef218b55
commit 5002175e80
2 changed files with 12 additions and 0 deletions
--- a/tests/lora/test_punica.py
+++ b/tests/lora/test_punica.py
@@ -53,6 +53,7 @@ H1 = H2 = [
    1152,
    1280,
    1536,
+    1664,
    2048,
    2304,
    2560,
@@ -66,6 +67,7 @@ H1 = H2 = [
    5120,
    5504,
    5632,
+    5888,
    6144,
    6400,
    6848,
@@ -75,10 +77,12 @@ H1 = H2 = [
    9216,
    10240,
    11008,
+    11264,
    13824,
    14336,
    15360,
    22016,
+    22528,
    24576,
    27392,
    27648,