Test (graph/quantize) Added extra prefix test to layerwise_quantize

Xilinx · Sep 9, 2024 · b3099bd · b3099bd
1 parent b714943
commit b3099bd
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/brevitas/graph/test_quantize.py b/tests/brevitas/graph/test_quantize.py
@@ -17,6 +17,11 @@
             'name_blacklist': ['0'],
             'key': '0',
             'expected': "<class 'torch.nn.modules.linear.Linear'>"},
+        {
+            'model': nn.Sequential(nn.Sequential(nn.Linear(2, 3))),
+            'name_blacklist': ['0'],
+            'key': '0.0',
+            'expected': "<class 'torch.nn.modules.linear.Linear'>"},
         {
             'model': nn.Sequential(nn.Sequential(nn.Linear(2, 3))),
             'name_blacklist': ['0.0'],