@@ -9,13 +9,16 @@ using Random
9
9
10
10
if ! haskey (ENV , " KA_BACKEND" )
11
11
const BACKEND = CPU ()
12
+ const Ts = (Float32, Float64)
12
13
else
13
14
backend = ENV [" KA_BACKEND" ]
14
15
if backend == " CPU"
15
16
const BACKEND = CPU ()
17
+ const Ts = (Float32, Float64)
16
18
elseif backend == " CUDA"
17
19
using CUDA
18
20
const BACKEND = CUDABackend ()
21
+ const Ts = (Float16, Float32, Float64)
19
22
else
20
23
error (" Backend $backend not recognized" )
21
24
end
31
34
SUITE[" saxpy" ] = BenchmarkGroup ()
32
35
33
36
let static = BenchmarkGroup ()
34
- for T in (Float16, Float32, Float64)
37
+ for T in Ts
35
38
dtype = BenchmarkGroup ()
36
39
for N in (64 , 256 , 512 , 1024 , 2048 , 4096 , 16384 , 32768 , 65536 , 262144 , 1048576 )
37
40
dtype[N] = @benchmarkable begin
@@ -49,7 +52,7 @@ let static = BenchmarkGroup()
49
52
end
50
53
51
54
let default = BenchmarkGroup ()
52
- for T in (Float16, Float32, Float64)
55
+ for T in Ts
53
56
dtype = BenchmarkGroup ()
54
57
for N in (64 , 256 , 512 , 1024 , 2048 , 4096 , 16384 , 32768 , 65536 , 262144 , 1048576 )
55
58
dtype[N] = @benchmarkable begin
0 commit comments