Skip to content

Commit 2a84ed5

Browse files
committed
Collect outputs during compare to support scalar values.
1 parent f3db760 commit 2a84ed5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/testsuite.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ function compare(f, AT::Type{<:GPUArray}, xs...)
2222
gpu_in = convert_array.(AT, xs)
2323
cpu_out = f(cpu_in...)
2424
gpu_out = f(gpu_in...)
25-
cpu_out Array(gpu_out)
25+
collect(cpu_out) collect(gpu_out)
2626
end
2727

2828
function supported_eltypes()

0 commit comments

Comments
 (0)