Skip to content

Commit ee591d7

Browse files
committed
flashattention_2: Take measurements on second Tc tile
1 parent 6fb6264 commit ee591d7

File tree

7 files changed

+13
-13
lines changed

7 files changed

+13
-13
lines changed

target/snitch_cluster/sw/apps/dnn/flashattention_2/tcasai/cfg/fp16-opt-vit-base.json

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@
33
// SPDX-License-Identifier: Apache-2.0
44

55
{
6-
L: 64,
7-
S: 64,
6+
L: 32,
7+
S: 128,
88
d: 64,
99
B_r: 16,
1010
B_c: 64,

target/snitch_cluster/sw/apps/dnn/flashattention_2/tcasai/cfg/fp16-opt-vit-huge.json

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@
33
// SPDX-License-Identifier: Apache-2.0
44

55
{
6-
L: 64,
7-
S: 64,
6+
L: 32,
7+
S: 128,
88
d: 80,
99
B_r: 16,
1010
B_c: 64,

target/snitch_cluster/sw/apps/dnn/flashattention_2/tcasai/cfg/fp32-opt-vit-base.json

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@
33
// SPDX-License-Identifier: Apache-2.0
44

55
{
6-
L: 64,
7-
S: 64,
6+
L: 32,
7+
S: 128,
88
d: 64,
99
B_r: 16,
1010
B_c: 64,

target/snitch_cluster/sw/apps/dnn/flashattention_2/tcasai/cfg/fp32-opt-vit-huge.json

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@
33
// SPDX-License-Identifier: Apache-2.0
44

55
{
6-
L: 64,
7-
S: 64,
6+
L: 32,
7+
S: 128,
88
d: 80,
99
B_r: 16,
1010
B_c: 64,

target/snitch_cluster/sw/apps/dnn/flashattention_2/tcasai/cfg/fp8-opt-vit-base.json

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@
33
// SPDX-License-Identifier: Apache-2.0
44

55
{
6-
L: 64,
7-
S: 64,
6+
L: 32,
7+
S: 128,
88
d: 64,
99
B_r: 16,
1010
B_c: 64,

target/snitch_cluster/sw/apps/dnn/flashattention_2/tcasai/cfg/fp8-opt-vit-huge.json

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@
33
// SPDX-License-Identifier: Apache-2.0
44

55
{
6-
L: 64,
7-
S: 64,
6+
L: 32,
7+
S: 128,
88
d: 80,
99
B_r: 16,
1010
B_c: 64,

target/snitch_cluster/sw/apps/dnn/flashattention_2/tcasai/results.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,7 @@ def get_total_runtime(experiment):
132132
# Define the simulated tc and tr iterations to extract the region runtimes from.
133133
# The second tc iteration is the worst case, since upon the first iteration
134134
# several calculations are not performed.
135-
tc_iteration = 0
135+
tc_iteration = 1
136136
tr_iteration = 0
137137

138138
# Calculate total runtime

0 commit comments

Comments
 (0)