docs: remove kernel example

LuxDL · Jan 24, 2025 · 49cd7be · 49cd7be
1 parent 06fe321
commit 49cd7be
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 94 deletions.
diff --git a/docs/src/.vitepress/config.mts b/docs/src/.vitepress/config.mts
@@ -365,10 +365,6 @@ export default defineConfig({
                 text: "Distributed Training",
                 link: "/manual/distributed_utils",
               },
-              {
-                text: "Lux In GPU Kernels",
-                link: "/manual/nn_inside_gpu_kernels",
-              },
             ],
           },
         ],

diff --git a/docs/src/manual/nn_inside_gpu_kernels.md b/docs/src/manual/nn_inside_gpu_kernels.md
diff --git a/lib/LuxLib/src/utils.jl b/lib/LuxLib/src/utils.jl
@@ -208,7 +208,7 @@ expand_batchdim(x::SVector{L, T}) where {L, T} = SMatrix{L, 1, T}(x)
 
 function CRC.rrule(::typeof(expand_batchdim), x::AbstractMatrix)
     ∇expand_batchdim = @closure Δ -> begin
-        return ∂∅, CRC.@thunk(CRC.ProjectTo(x)(proj_x(view(recursive_unthunk(Δ), :, :, 1))))
+        return ∂∅, CRC.@thunk(CRC.ProjectTo(x)(view(recursive_unthunk(Δ), :, :, 1)))
     end
     return expand_batchdim(x), ∇expand_batchdim
 end