Skip to content

Commit 186a50d

Browse files
committed
Generalize supports_autoenzyme from utils to Domains
1 parent 1549195 commit 186a50d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/utils.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ end
1414

1515
"""
1616
supports_autoenzyme(geometry::Geometry)
17-
supports_autoenzyme(type::Type{<:Geometry})
17+
supports_autoenzyme(type::Type{<:GeometryOrDomain})
1818
1919
Return whether a geometry or geometry type has a parametric function that can be
2020
differentiated with Enzyme. See GitHub Issue #154 for more information.
@@ -25,7 +25,7 @@ function supports_autoenzyme end
2525
supports_autoenzyme(::Type{<:Any}) = false
2626

2727
# If provided a geometry instance, re-run with the type as argument
28-
supports_autoenzyme(::G) where {G <: Geometry} = supports_autoenzyme(G)
28+
supports_autoenzyme(::G) where {G <: Meshes.GeometryOrDomain} = supports_autoenzyme(G)
2929

3030
"""
3131
_check_diff_method_support(::Geometry, ::DifferentiationMethod) -> nothing

0 commit comments

Comments
 (0)