Skip to content

Commit bd24a5e

Browse files
committed
Added Demo TinyViT FP test for Siracusa Tiled. Updated test L1 limits for new constraint set. Bugifx for input/output L3 buffer name check, where similar naming of internal nodes would confuse the dumper
1 parent eecf135 commit bd24a5e

File tree

2 files changed

+24
-17
lines changed

2 files changed

+24
-17
lines changed

.github/workflows/ci-platform-siracusa-tiled.yml

Lines changed: 18 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -48,13 +48,13 @@ jobs:
4848
{"name":"RQHardswish","L1":[750]},
4949
{"name":"testFloatGEMM","L1":[8000]},
5050
51-
{"name":"testFloat2DConvolution","L1":[1200]},
52-
{"name":"testFloat2DConvolutionBias","L1":[6000]},
53-
{"name":"testFloat2DConvolutionZeroBias","L1":[6000]},
51+
{"name":"testFloat2DConvolution","L1":[1600]},
52+
{"name":"testFloat2DConvolutionBias","L1":[6600]},
53+
{"name":"testFloat2DConvolutionZeroBias","L1":[6600]},
5454
55-
{"name":"testFloat2DDWConvolution","L1":[6600]},
56-
{"name":"testFloat2DDWConvolutionBias","L1":[6600]},
57-
{"name":"testFloat2DDWConvolutionZeroBias","L1":[6600]},
55+
{"name":"testFloat2DDWConvolution","L1":[7200]},
56+
{"name":"testFloat2DDWConvolutionBias","L1":[7200]},
57+
{"name":"testFloat2DDWConvolutionZeroBias","L1":[7200]},
5858
5959
{"name":"testFloatLayerNorm","L1":[2000]},
6060
{"name":"testFloatMaxPool","L1":[2000]},
@@ -88,13 +88,13 @@ jobs:
8888
{"name":"RQHardswish","L1":[800]},
8989
{"name":"testFloatGEMM","L1":[8000]},
9090
91-
{"name":"testFloat2DConvolution","L1":[1600]},
92-
{"name":"testFloat2DConvolutionBias","L1":[8000]},
93-
{"name":"testFloat2DConvolutionZeroBias","L1":[8000]},
91+
{"name":"testFloat2DConvolution","L1":[2000]},
92+
{"name":"testFloat2DConvolutionBias","L1":[8800]},
93+
{"name":"testFloat2DConvolutionZeroBias","L1":[8800]},
9494
95-
{"name":"testFloat2DDWConvolution","L1":[9000]},
96-
{"name":"testFloat2DDWConvolutionBias","L1":[9000]},
97-
{"name":"testFloat2DDWConvolutionZeroBias","L1":[9000]},
95+
{"name":"testFloat2DDWConvolution","L1":[9800]},
96+
{"name":"testFloat2DDWConvolutionBias","L1":[10000]},
97+
{"name":"testFloat2DDWConvolutionZeroBias","L1":[9800]},
9898
9999
{"name":"testFloatLayerNorm","L1":[2000]},
100100
{"name":"testFloatMaxPool","L1":[5000]},
@@ -138,6 +138,8 @@ jobs:
138138
L1: [2000, 64000]
139139
- name: "testTrainCCT/CCT1_Classifier_Training/CCT_1_16_16_8"
140140
L1: [4000, 64000]
141+
- name: "testFloatDemoTinyViT"
142+
L1: [4000]
141143
num-cores: [8]
142144
uses: ./.github/workflows/_runner-siracusa-tiled.yml
143145
with:
@@ -169,6 +171,8 @@ jobs:
169171
L1: [64000, 128000]
170172
- name: "testTrainCCT/CCT1_Classifier_Training/CCT_1_16_16_128"
171173
L1: [32000, 64000]
174+
- name: "testFloatDemoTinyViT"
175+
L1: [4000]
172176
num-cores: [8]
173177
default-memory-level: ["L3"]
174178
uses: ./.github/workflows/_runner-siracusa-tiled.yml
@@ -207,6 +211,8 @@ jobs:
207211
L1: [64000, 128000]
208212
- name: "testTrainCCT/CCT1_Classifier_Training/CCT_1_16_16_128"
209213
L1: [8000, 64000]
214+
- name: "testFloatDemoTinyViT"
215+
L1: [4000]
210216
num-cores: [8]
211217
double-buffer: [true]
212218
default-memory-level: ["L3"]

DeeployTest/testUtils/codeGenerate.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -224,13 +224,14 @@ def type2TypeStr(dataType) -> Tuple[str, int]:
224224

225225
def dumpBuffer(buf: VariableBuffer, path: str):
226226

227-
if "input" in buf.name:
228-
idx = int(buf.name.split("_")[1])
227+
# Check if buffer name matches exactly "input_N" or "output_N" pattern
228+
parts = buf.name.split("_")
229+
if len(parts) == 2 and parts[0] == "input" and parts[1].isdigit():
230+
idx = int(parts[1])
229231
array = _shapeBroadcast(deployer.ctxt, test_inputs[idx], f"input_{idx}")
230232

231-
elif "output" in buf.name:
232-
_list = buf.name.split("_")
233-
idx = int(_list[1])
233+
elif len(parts) == 2 and parts[0] == "output" and parts[1].isdigit():
234+
idx = int(parts[1])
234235
array = _shapeBroadcast(deployer.ctxt, test_outputs[idx], f"output_{idx}")
235236

236237
elif isinstance(buf, ConstantBuffer):

0 commit comments

Comments
 (0)