Skip to content

Commit 8c8b85a

Browse files
Merge pull request #1897 from CliMA/ck/update_bm_script
Small update to benchmark script
2 parents 578b992 + c555bf4 commit 8c8b85a

File tree

1 file changed

+19
-19
lines changed

1 file changed

+19
-19
lines changed

benchmarks/scripts/indexing_and_static_ndranges.jl

Lines changed: 19 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -42,17 +42,17 @@ at_dot_call!($X_array, $Y_array):
4242
143 milliseconds, 774 microseconds
4343
at_dot_call!($X_vector, $Y_vector):
4444
65 milliseconds, 567 microseconds
45-
custom_kernel_bc!($X_vector, $Y_vector, $us; printtb = false):
45+
custom_kernel_bc!($X_vector, $Y_vector, $us):
4646
66 milliseconds, 870 microseconds
47-
custom_kernel_bc!($X_array, $Y_array, $us; printtb = false, use_pw = false):
47+
custom_kernel_bc!($X_array, $Y_array, $us; use_pw = false):
4848
143 milliseconds, 643 microseconds
49-
custom_kernel_bc!($X_array, $Y_array, $us; printtb = false, use_pw = true):
49+
custom_kernel_bc!($X_array, $Y_array, $us; use_pw = true):
5050
65 milliseconds, 778 microseconds
51-
custom_kernel_bc!($X_vector, $Y_vector, $uss; printtb = false):
51+
custom_kernel_bc!($X_vector, $Y_vector, $uss):
5252
65 milliseconds, 765 microseconds
53-
custom_kernel_bc!($X_array, $Y_array, $uss; printtb = false, use_pw = false):
53+
custom_kernel_bc!($X_array, $Y_array, $uss; use_pw = false):
5454
144 milliseconds, 271 microseconds
55-
custom_kernel_bc!($X_array, $Y_array, $uss; printtb = false, use_pw = true):
55+
custom_kernel_bc!($X_array, $Y_array, $uss; use_pw = true):
5656
66 milliseconds, 376 microseconds
5757
```
5858
@@ -64,17 +64,17 @@ at_dot_call!($X_vector, $Y_vector):
6464
2 milliseconds, 834 microseconds
6565
custom_sol_kernel!($X_vector, $Y_vector, $(Val(N))):
6666
2 milliseconds, 547 microseconds
67-
custom_kernel_bc!($X_vector, $Y_vector, $us; printtb = false):
67+
custom_kernel_bc!($X_vector, $Y_vector, $us):
6868
2 milliseconds, 561 microseconds
69-
custom_kernel_bc!($X_array, $Y_array, $us; printtb = false, use_pw = false):
69+
custom_kernel_bc!($X_array, $Y_array, $us; use_pw = false):
7070
4 milliseconds, 160 microseconds
71-
custom_kernel_bc!($X_array, $Y_array, $us; printtb = false, use_pw = true):
71+
custom_kernel_bc!($X_array, $Y_array, $us; use_pw = true):
7272
2 milliseconds, 584 microseconds
73-
custom_kernel_bc!($X_vector, $Y_vector, $uss; printtb = false):
73+
custom_kernel_bc!($X_vector, $Y_vector, $uss):
7474
2 milliseconds, 540 microseconds
75-
custom_kernel_bc!($X_array, $Y_array, $uss; printtb = false, use_pw = false):
75+
custom_kernel_bc!($X_array, $Y_array, $uss; use_pw = false):
7676
2 milliseconds, 715 microseconds
77-
custom_kernel_bc!($X_array, $Y_array, $uss; printtb = false, use_pw = true):
77+
custom_kernel_bc!($X_array, $Y_array, $uss; use_pw = true):
7878
2 milliseconds, 547 microseconds
7979
```
8080
=#
@@ -319,7 +319,7 @@ us_tup = (1, 2, 3)
319319
@test get_Nh(UniversalSizesCC(us_tup...)) == get_Nh(UniversalSizesStatic(us_tup...))
320320
@test get_N(UniversalSizesCC(us_tup...)) == get_N(UniversalSizesStatic(us_tup...))
321321

322-
function custom_kernel_bc!(X, Y, us::AbstractUniversalSizes; printtb=true, use_pw=true)
322+
function custom_kernel_bc!(X, Y, us::AbstractUniversalSizes; printtb=false, use_pw=true)
323323
(; x1, x2, x3) = X
324324
(; y1) = Y
325325
bc_base = @lazy @. y1 = myadd(x1, x2, x3)
@@ -399,12 +399,12 @@ custom_kernel_bc!(X_array, Y_array, uss; use_pw=true)
399399
@pretty_belapsed at_dot_call!($X_array, $Y_array) # slow
400400
@pretty_belapsed at_dot_call!($X_vector, $Y_vector) # fast
401401
iscpu || @pretty_belapsed custom_sol_kernel!($X_vector, $Y_vector, $(Val(N)))
402-
@pretty_belapsed custom_kernel_bc!($X_vector, $Y_vector, $us; printtb=false)
403-
@pretty_belapsed custom_kernel_bc!($X_array, $Y_array, $us; printtb=false, use_pw=false)
404-
@pretty_belapsed custom_kernel_bc!($X_array, $Y_array, $us; printtb=false, use_pw=true)
402+
@pretty_belapsed custom_kernel_bc!($X_vector, $Y_vector, $us)
403+
@pretty_belapsed custom_kernel_bc!($X_array, $Y_array, $us; use_pw=false)
404+
@pretty_belapsed custom_kernel_bc!($X_array, $Y_array, $us; use_pw=true)
405405

406-
@pretty_belapsed custom_kernel_bc!($X_vector, $Y_vector, $uss; printtb=false)
407-
@pretty_belapsed custom_kernel_bc!($X_array, $Y_array, $uss; printtb=false, use_pw=false)
408-
@pretty_belapsed custom_kernel_bc!($X_array, $Y_array, $uss; printtb=false, use_pw=true)
406+
@pretty_belapsed custom_kernel_bc!($X_vector, $Y_vector, $uss)
407+
@pretty_belapsed custom_kernel_bc!($X_array, $Y_array, $uss; use_pw=false)
408+
@pretty_belapsed custom_kernel_bc!($X_array, $Y_array, $uss; use_pw=true)
409409

410410
#! format: on

0 commit comments

Comments
 (0)