Make Documenter happy

Evovest · Oct 5, 2023 · 1c391a8 · 1c391a8
1 parent 437b27d
commit 1c391a8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 25 deletions.
diff --git a/ext/EvoTreesCUDAExt/fit-utils.jl b/ext/EvoTreesCUDAExt/fit-utils.jl
@@ -1,6 +1,3 @@
-"""
-    hist_kernel!
-"""
 function hist_kernel!(h∇::CuDeviceArray{T,3}, ∇::CuDeviceMatrix{S}, x_bin, is, js) where {T,S}
     tix, tiy, k = threadIdx().z, threadIdx().y, threadIdx().x
     bdx, bdy = blockDim().z, blockDim().y
@@ -48,9 +45,6 @@ function update_hist_gpu!(h, h∇, ∇, x_bin, is, js, jsc)
     return nothing
 end
 
-"""
-    hist_kernel_vec!
-"""
 function hist_kernel_vec!(h∇, ∇, x_bin, is)
     tix, k = threadIdx().x, threadIdx().y
     bdx = blockDim().x
@@ -103,10 +97,8 @@ function update_hist_gpu_vec!(h, h∇, ∇, x_bin, is, js::Vector)
     return nothing
 end
 
-"""
-    Multi-threads split_set!
-        Take a view into left and right placeholders. Right ids are assigned at the end of the length of the current node set.
-"""
+# Multi-threads split_set!
+# Take a view into left and right placeholders. Right ids are assigned at the end of the length of the current node set.
 function split_chunk_kernel!(
     left::CuDeviceVector{S},
     right::CuDeviceVector{S},

diff --git a/ext/EvoTreesCUDAExt/predict.jl b/ext/EvoTreesCUDAExt/predict.jl
@@ -1,6 +1,3 @@
-"""
-    predict_kernel!
-"""
 function predict_kernel!(
     ::Type{L},
     pred::CuDeviceMatrix{T},
@@ -28,10 +25,7 @@ function predict_kernel!(
     return nothing
 end
 
-"""
-    predict_kernel!
-        GradientRegression
-"""
+# GradientRegression
 function predict_kernel!(
     ::Type{<:EvoTrees.GradientRegression},
     pred::CuDeviceMatrix{T},
@@ -56,10 +50,7 @@ function predict_kernel!(
     return nothing
 end
 
-"""
-    predict_kernel!
-        Logistic
-"""
+# Logistic
 function predict_kernel!(
     ::Type{<:EvoTrees.LogLoss},
     pred::CuDeviceMatrix{T},
@@ -84,10 +75,7 @@ function predict_kernel!(
     return nothing
 end
 
-"""
-    predict_kernel!
-        MLE2P
-"""
+# MLE2P
 function predict_kernel!(
     ::Type{<:EvoTrees.MLE2P},
     pred::CuDeviceMatrix{T},