@@ -59,12 +59,18 @@ gpu_mat_ints = MtlMatrix{Int64}(rand(rng, -10:10, m, n))
5959gpu_mat_long_ints = MtlMatrix {Int64} (rand (rng, - 10 : 10 , m_long, n_long))
6060gpu_vec_ints = reshape (gpu_mat_ints, length (gpu_mat_ints))
6161
62- # let group = addgroup!(group, "reverse")
63- # group["1d"] = @benchmarkable Metal.@sync reverse($gpu_vec)
64- # group["2d"] = @benchmarkable Metal.@sync reverse($gpu_mat; dims=1)
65- # group["1d_inplace"] = @benchmarkable Metal.@sync reverse!($gpu_vec)
66- # group["2d_inplace"] = @benchmarkable Metal.@sync reverse!($gpu_mat; dims=1)
67- # end
62+ let group = addgroup! (group, " reverse" )
63+ group[" 1d" ] = @benchmarkable Metal. @sync reverse ($ gpu_vec)
64+ group[" dims=1" ] = @benchmarkable Metal. @sync reverse ($ gpu_mat; dims= 1 )
65+ group[" dims=2" ] = @benchmarkable Metal. @sync reverse ($ gpu_mat; dims= 2 )
66+ group[" dims=1L" ] = @benchmarkable Metal. @sync reverse ($ gpu_mat_long; dims= 1 )
67+ group[" dims=2L" ] = @benchmarkable Metal. @sync reverse ($ gpu_mat_long; dims= 2 )
68+ group[" 1d_inplace" ] = @benchmarkable Metal. @sync reverse! ($ gpu_vec)
69+ group[" dims=1_inplace" ] = @benchmarkable Metal. @sync reverse! ($ gpu_mat; dims= 1 )
70+ group[" dims=2_inplace" ] = @benchmarkable Metal. @sync reverse! ($ gpu_mat; dims= 2 )
71+ group[" dims=1L_inplace" ] = @benchmarkable Metal. @sync reverse! ($ gpu_mat_long; dims= 1 )
72+ group[" dims=2L_inplace" ] = @benchmarkable Metal. @sync reverse! ($ gpu_mat_long; dims= 2 )
73+ end
6874
6975# 'evals=1' added to prevent hang when running benchmarks of CI
7076# TODO : Investigate cause and properly fix.
0 commit comments