Skip to content

Commit

Permalink
fix typo in togpu
Browse files Browse the repository at this point in the history
  • Loading branch information
liuanji committed Feb 15, 2021
1 parent 8bf4f32 commit 31d1c90
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/Utils/data.jl
Original file line number Diff line number Diff line change
Expand Up @@ -317,7 +317,7 @@ to_gpu(m::Array) = CuArray(m)
to_gpu(v::BitVector) =
AbstractBitVector(to_gpu(chunks(v)), length(v))
to_gpu(v::Vector{Union{F,Missing}}) where F<:AbstractFloat =
CuArray(T.(coalesce(v,typemax(T))))
CuArray(F.(coalesce(v,typemax(F))))
to_gpu(v::Vector{Union{Bool,Missing}}) =
CuArray(UInt8.(coalesce.(v,typemax(UInt8))))
to_gpu(df::DataFrame) = isgpu(df) ? df : mapcols(to_gpu, df)
Expand Down

0 comments on commit 31d1c90

Please sign in to comment.