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

chore: trigger build with latest Lux #882

Open
wants to merge 14 commits into
base: master
Choose a base branch
from
Open

chore: trigger build with latest Lux #882

wants to merge 14 commits into from

Conversation

avik-pal
Copy link
Member

No description provided.

@avik-pal avik-pal force-pushed the ap/trigger branch 6 times, most recently from 361e8ec to 0c724f7 Compare August 21, 2024 01:48
@avik-pal
Copy link
Member Author

avik-pal commented Aug 21, 2024

@ChrisRackauckas I tracked down and fixed the upstream issues. The only one remaining is https://github.com/SciML/NeuralPDE.jl/actions/runs/10485955321/job/29043159135?pr=882#step:6:1216 but that is kind of hard to handle -- it is matrix multiply with non-concrete element types.

Might be best to get the adapt usage fixed on NeuralPDE end that will fix all the issues

@sathvikbhagavan
Copy link
Member

Might be best to get the adapt usage fixed on NeuralPDE end that will fix all the issues

I can look into it. Can you open an issue about it?

Also, in CI, why is precompilation failing - https://github.com/SciML/NeuralPDE.jl/actions/runs/10485955321/job/29043160310?pr=882#step:6:982

@avik-pal
Copy link
Member Author

avik-pal commented Aug 21, 2024

I can look into it. Can you open an issue about it?

#883

Also, in CI, why is precompilation failing - SciML/NeuralPDE.jl/actions/runs/10485955321/job/29043160310?pr=882#step:6:982

Depwarn errors are on, and currently Enzyme doesn't support the latest GPUCompiler which doesn't have those depwarns (same as SciML/DeepEquilibriumNetworks.jl#159)

@ChrisRackauckas
Copy link
Member

Might be best to get the adapt usage fixed on NeuralPDE end that will fix all the issues

That's the right thing to do.

@ChrisRackauckas
Copy link
Member

@wsmoses what should we do about the GPUCompiler Enzyme depwarn issue? How close is that to getting the release?

@wsmoses
Copy link

wsmoses commented Aug 26, 2024

Enzyme released a version with the latest gpucompiler last night

@ChrisRackauckas
Copy link
Member

@avik-pal you're unblocked here?

@avik-pal
Copy link
Member Author

avik-pal commented Sep 1, 2024

No I need #883 to be fixed, else it tries to propagate Matrix{Any}

@ChrisRackauckas
Copy link
Member

Okay, but at least unblocked from non-NeuralPDE things 😅 . @KirillZubov do you think you can have a look at that?

@KirillZubov
Copy link
Member

@ChrisRackauckas ok, I will try figure out with it , it's a little related to my current task

@@ -217,20 +217,20 @@ function generate_training_sets(domains, dx, eqs, bcs, eltypeθ, dict_indvars::D

bcs_train_sets = map(bound_args) do bt
span = map(b -> get(dict_var_span, b, b), bt)
_set = adapt(eltypeθ,
_set = adapt(Array{eltypeθ},
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Do not force an Array type, since that won't work in most scenarios. Just add a post-adapt convert.(eltypeθ, x)

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.

5 participants