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.
2 parents 0ea250a + 3900814 commit 1b42b98Copy full SHA for 1b42b98
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