diff --git a/julia/deps/build.jl b/julia/deps/build.jl index a87343d9dab5..a79d2a062c18 100644 --- a/julia/deps/build.jl +++ b/julia/deps/build.jl @@ -54,7 +54,7 @@ if Sys.isunix() nvcc_path = Sys.which("nvcc") if nvcc_path ≢ nothing @info "Found nvcc: $nvcc_path" - push!(CUDAPATHS, replace(nvcc_path, "bin/nvcc", "lib64")) + push!(CUDAPATHS, replace(nvcc_path, "bin/nvcc" => "lib64")) end end diff --git a/julia/src/metric.jl b/julia/src/metric.jl index f1cdc68d947f..2ae7fc85144b 100644 --- a/julia/src/metric.jl +++ b/julia/src/metric.jl @@ -260,7 +260,7 @@ end function get(metric::MSE) # Delay copy until last possible moment - mse_sum = mapreduce(nda->copy(nda)[1], +, 0.0, metric.mse_sum) + mse_sum = mapreduce(nda->copy(nda)[1], +, metric.mse_sum; init = zero(MX_float)) [(:MSE, mse_sum / metric.n_sample)] end