Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix union abi #1150

Merged
merged 3 commits into from
Nov 18, 2023
Merged

Fix union abi #1150

merged 3 commits into from
Nov 18, 2023

Conversation

wsmoses
Copy link
Member

@wsmoses wsmoses commented Nov 18, 2023

I have failed to make an enzyme indep example, but this is the trigger:

using Oceananigans
using Enzyme

using Oceananigans.Fields

Enzyme.API.printall!(true)

struct MyFlux <: Oceananigans.BoundaryConditions.AbstractBoundaryConditionClassification end

args = (
	Val{(false, false, false)},
	Val(1),
	Val((true, true, true)),
	Base.Val(NamedTuple{(Symbol("1"), Symbol("2"), Symbol("3")), Tuple{Any, Any, Any}}),
	Base.getindex,
	nothing,
	((nothing,), Oceananigans.BoundaryConditions.BoundaryCondition(MyFlux(),nothing)),
	((nothing,), Oceananigans.BoundaryConditions.BoundaryCondition(MyFlux(),nothing)),
	1,
	nothing
)

using InteractiveUtils
@show InteractiveUtils.@code_typed Enzyme.Compiler.runtime_generic_augfwd(args...)
@show InteractiveUtils.@code_llvm dump_module=true Enzyme.Compiler.runtime_generic_augfwd(args...)
@show InteractiveUtils.@code_native Enzyme.Compiler.runtime_generic_augfwd(args...)
Enzyme.Compiler.runtime_generic_augfwd(args...)

@wsmoses wsmoses requested a review from vchuravy November 18, 2023 01:39
@wsmoses wsmoses merged commit 9579ec3 into main Nov 18, 2023
28 of 42 checks passed
@wsmoses wsmoses deleted the unionabi branch November 18, 2023 02:47
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant