[ci] Extend tests with no stalls

pulp-platform · Oct 10, 2024 · 4d3294e · 4d3294e
1 parent 3786b6a
commit 4d3294e
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 3 deletions.
diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
@@ -45,19 +45,34 @@ run_sim:
       P: 64
       F: 64
       activation: gelu
+      no_stalls: 0
+    - S: 64
+      E: 64
+      P: 64
+      F: 64
+      activation: gelu
+      no_stalls: 1
+    - S: 128
+      E: 192
+      P: 256
+      F: 256
+      activation: gelu
+      no_stalls: 0
     - S: 128
       E: 192
       P: 256
       F: 256
       activation: gelu
+      no_stalls: 1
     - S: 192
       E: 256
       P: 128
       F: 128
       activation: relu
+      no_stalls: 1
   script:
     - make bender
-    - make sim VSIM_FLAGS=-c s=$S e=$E p=$P f=$F bias=1 activation=$activation
+    - make sim VSIM_FLAGS=-c s=$S e=$E p=$P f=$F bias=1 activation=$activation no_stalls=$no_stalls
     - ./modelsim/return_status.sh modelsim/build/transcript $S $E $P $F ita_tb
 
 run_hwpe_sim:
@@ -71,17 +86,32 @@ run_hwpe_sim:
       P: 64
       F: 64
       activation: gelu
+      no_stalls: 0
+    - S: 64
+      E: 64
+      P: 64
+      F: 64
+      activation: gelu
+      no_stalls: 1
+    - S: 128
+      E: 192
+      P: 256
+      F: 256
+      activation: gelu
+      no_stalls: 0
     - S: 128
       E: 192
       P: 256
       F: 256
       activation: gelu
+      no_stalls: 1
     - S: 192
       E: 256
       P: 128
       F: 128
       activation: relu
+      no_stalls: 1
   script:
     - make bender
-    - make sim VSIM_FLAGS=-c DEBUG=OFF target=sim_ita_hwpe_tb s=$S e=$E p=$P f=$F bias=1 activation=$activation
+    - make sim VSIM_FLAGS=-c DEBUG=OFF target=sim_ita_hwpe_tb s=$S e=$E p=$P f=$F bias=1 activation=$activation no_stalls=$no_stalls
     - ./modelsim/return_status.sh modelsim/build/transcript $S $E $P $F hwpe_tb
diff --git a/src/hwpe/tb/ita_hwpe_tb.sv b/src/hwpe/tb/ita_hwpe_tb.sv
@@ -67,7 +67,7 @@ module ita_hwpe_tb;
 
   // HWPE Parameters
   localparam unsigned ITA_REG_OFFSET  = 32'h20;
-  parameter real PROB_STALL = 0.1;
+  parameter real PROB_STALL = `ifdef NO_STALLS ((`NO_STALLS == 1) ? 0 : 0.1) `else 0.1 `endif;
   parameter MEMORY_SIZE = SEQUENCE_LEN*EMBEDDING_SIZE*4+EMBEDDING_SIZE*PROJECTION_SPACE*4+PROJECTION_SPACE*3*3+EMBEDDING_SIZE*3+SEQUENCE_LEN*PROJECTION_SPACE*4+SEQUENCE_LEN*SEQUENCE_LEN+EMBEDDING_SIZE*FEEDFORWARD_SIZE*2+FEEDFORWARD_SIZE*3+EMBEDDING_SIZE*3;
 
   parameter int unsigned AccDataWidth = ITA_TCDM_DW;