You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
using LinearAlgebra
using BenchmarkTools
using Random
using CUDA
using Test
@testset"dot $atype"for atype in [Array, CuArray]
Random.seed!(100)
N =10^2
A =atype(rand(ComplexF64, N,N))
B = [A]
C = [B]
D = [C]
@btime CUDA.@syncdot($A, $A)
@btime CUDA.@syncdot($B, $B)
@btime CUDA.@syncdot($C, $C)
@btime CUDA.@syncdot($D, $D)
end1.400 μs (0 allocations:0 bytes)
2.522 μs (0 allocations:0 bytes)
4.743 μs (0 allocations:0 bytes)
9.200 μs (0 allocations:0 bytes)
Test Summary:| Total Time
dot Array |013.9s
84.900 μs (14 allocations:240 bytes)
207.000 μs (28 allocations:480 bytes)
874.700 μs (56 allocations:960 bytes)
2.081 ms (112 allocations:1.88 KiB)
Test Summary:| Total Time
dot CuArray |044.5s
The Minimal Working Example (MWE) for this bug:
Manifest.toml
Expected behavior
A clear and concise description of what you expected to happen.
Version info
Details on Julia:
Details on CUDA:
The text was updated successfully, but these errors were encountered: