fix: mark Utils.eltype as non-differentiable

LuxDL · Sep 5, 2024 · 7b96d5d · 7b96d5d
1 parent dce920d
commit 7b96d5d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/docs/src/introduction/updating_to_v1.md b/docs/src/introduction/updating_to_v1.md
@@ -95,7 +95,7 @@ abstraction.
 - `Lux.Experimental.@layer_map` is not longer needed and has been removed. The name of the
  variable prevents writing generic functions and is no longer pre-pended to the `KeyPath`.
  See the docstring of [`Lux.Experimental.layer_map`](@ref) for more details.
-- `allow_fast_activation` kwarg has been removed completely. Pass an annonymous function
+- `allow_fast_activation` kwarg has been removed completely. Pass an anonymous function
  as the activation to prevent internal modivations to the activation function.
 
 ### Breaking Changes (Moved Functionality)

diff --git a/src/utils.jl b/src/utils.jl
@@ -126,6 +126,8 @@ eltype(x) = eltype(Base.eltype(x))
 eltype(::Type{T}) where {T} = T
 eltype(::Type{<:Dual{T, V}}) where {T, V} = V
 
+@non_differentiable eltype(::Any)
+
 ofeltype_array(::Type{T}, x::AbstractArray) where {T} = broadcast(T, x)
 function ofeltype_array(::Type{T}, x::AbstractArray{<:Dual{Tag, V, N}}) where {Tag, T, V, N}
  return Dual{Tag, T, N}.(x)