From 2677796cf824d5ec6b0a812e3186d69ad534415d Mon Sep 17 00:00:00 2001 From: Matteo Perotti Date: Thu, 22 Feb 2024 10:08:43 +0100 Subject: [PATCH] [scripts] Fix performance script --- scripts/performance.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/scripts/performance.py b/scripts/performance.py index 8fe3bb433..ac98c8197 100755 --- a/scripts/performance.py +++ b/scripts/performance.py @@ -87,7 +87,7 @@ def exp(args, cycles): def softmax(args, cycles): channels = int(args[0]) insize = int(args[1]) - performance = 25 * channels * insize / cycles + performance = 34 * channels * insize / cycles return [insize, performance] def pathfinder(args, cycles): num_runs = int(args[0]) @@ -141,10 +141,10 @@ def roi_align(args, cycles): 'fconv3d' : lambda l, s : 2 * l * 8/s, 'jacobi2d' : lambda l, s : l * 8/s, 'dropout' : lambda l, s : 4 * l / (2*s + 1/8), - 'fft' : lambda l, s : 6/5 * l * 8/s, + 'fft' : lambda l, s : 5/4 * l * 8/s, 'dwt' : lambda l, s : 4 * l / s, - 'exp' : lambda l, s : 28/23 * l * 8/s, - 'softmax' : lambda l, s : 32/25 * l * 8/s, + 'exp' : lambda l, s : 30/23 * l * 8/s, + 'softmax' : lambda l, s : 34/27 * l * 8/s, 'pathfinder' : lambda l, s : l * 8/s, 'dotproduct' : lambda l, s : l * 8/s, 'fdotproduct': lambda l, s : l * 8/s, @@ -161,14 +161,14 @@ def roi_align(args, cycles): 'fconv3d' : lambda l, s : 2 * l * 8/s, 'jacobi2d' : lambda l, s : l * 8/s, 'dropout' : lambda l, s : 4 * l / (2*s + 1/8), - 'fft' : lambda l, s : 6/5 * l * 8/s, + 'fft' : lambda l, s : 5/4 * l * 8/s, 'dwt' : lambda l, s : 4 * l / s, - 'exp' : lambda l, s : 28/23 * l * 8/s, - 'softmax' : lambda l, s : 32/25 * l * 8/s, + 'exp' : lambda l, s : 30/23 * l * 8/s, + 'softmax' : lambda l, s : 34/27 * l * 8/s, 'pathfinder' : lambda l, s : l * 8/s, 'dotproduct' : lambda l, s : 4 * l/s, 'fdotproduct': lambda l, s : 4 * l/s, - 'roi_align' : lambda l, s : 3/5 * l * 8/s, + 'roi_align' : lambda l, s : 9/5 * l * 4/s, } def main():