@@ -9,13 +9,16 @@ using Random
99
1010if ! haskey (ENV , " KA_BACKEND" )
1111 const BACKEND = CPU ()
12+ const Ts = (Float32, Float64)
1213else
1314 backend = ENV [" KA_BACKEND" ]
1415 if backend == " CPU"
1516 const BACKEND = CPU ()
17+ const Ts = (Float32, Float64)
1618 elseif backend == " CUDA"
1719 using CUDA
1820 const BACKEND = CUDABackend ()
21+ const Ts = (Float16, Float32, Float64)
1922 else
2023 error (" Backend $backend not recognized" )
2124 end
3134SUITE[" saxpy" ] = BenchmarkGroup ()
3235
3336let static = BenchmarkGroup ()
34- for T in (Float16, Float32, Float64)
37+ for T in Ts
3538 dtype = BenchmarkGroup ()
3639 for N in (64 , 256 , 512 , 1024 , 2048 , 4096 , 16384 , 32768 , 65536 , 262144 , 1048576 )
3740 dtype[N] = @benchmarkable begin
@@ -49,7 +52,7 @@ let static = BenchmarkGroup()
4952end
5053
5154let default = BenchmarkGroup ()
52- for T in (Float16, Float32, Float64)
55+ for T in Ts
5356 dtype = BenchmarkGroup ()
5457 for N in (64 , 256 , 512 , 1024 , 2048 , 4096 , 16384 , 32768 , 65536 , 262144 , 1048576 )
5558 dtype[N] = @benchmarkable begin
0 commit comments