We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0ea250a commit 3900814Copy full SHA for 3900814
src/testsuite.jl
@@ -17,11 +17,11 @@ using StaticArrays
17
convert_array(f, x) = f(x)
18
convert_array(f, x::Base.RefValue) = x[]
19
20
-function compare(f, AT::Type{<:GPUArray}, xs...)
+function compare(f, AT::Type{<:GPUArray}, xs...; kwargs...)
21
cpu_in = convert_array.(copy, xs)
22
gpu_in = convert_array.(AT, xs)
23
- cpu_out = f(cpu_in...)
24
- gpu_out = f(gpu_in...)
+ cpu_out = f(cpu_in...; kwargs...)
+ gpu_out = f(gpu_in...; kwargs...)
25
collect(cpu_out) ≈ collect(gpu_out)
26
end
27
0 commit comments