We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7b3bd07 commit 26237ffCopy full SHA for 26237ff
1 file changed
src/host/broadcast.jl
@@ -47,6 +47,9 @@ end
47
@inline function _copyto!(dest::AbstractArray, bc::Broadcasted)
48
axes(dest) == axes(bc) || Broadcast.throwdm(axes(dest), axes(bc))
49
isempty(dest) && return dest
50
+ if eltype(dest) <: BrokenBroadcast
51
+ throw(ArgumentError("Broadcast operation resulting in $(eltype(eltype(dest))) is not GPU compatible"))
52
+ end
53
bc = Broadcast.preprocess(dest, bc)
54
55
@kernel function broadcast_kernel_linear(dest, bc)
@@ -69,10 +72,6 @@ end
69
72
70
73
# ndims check for 0D support
71
74
broadcast_kernel(dest, bc; ndrange = ndims(dest) > 0 ? size(dest) : (1,))
- if eltype(dest) <: BrokenBroadcast
- throw(ArgumentError("Broadcast operation resulting in $(eltype(eltype(dest))) is not GPU compatible"))
- end
75
-
76
return dest
77
end
78
0 commit comments