@@ -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