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

Remove NNlibCUDA dependency to support Flux 0.14 #248

Merged
merged 1 commit into from
Jul 17, 2023

Conversation

christiangnrd
Copy link
Contributor

This is in regards to the question I posted on the slack. I asked if NNlibCUDA was a necessary dependency for Metalhead since there is no direct reference to it in the code and it's preventing Flux 0.14 from being used in the same environment as Metalhead.

I made this a draft because I don't know the full repercussions of this change. I'm seeking feedback from more experienced Metalhead/Flux developers.

@CarloLucibello
Copy link
Member

Seems fine. Could CUDA as well be removed?

@christiangnrd
Copy link
Contributor Author

There are 2 dropblock_mask functions for cuda in src/layers/drop.jl.

I played around with putting them in an extension but there are design decisions that I am not comfortable making so I figured this would be a good solution for until a proper extension can be made.

@christiangnrd christiangnrd marked this pull request as ready for review July 16, 2023 22:38
@darsnack darsnack merged commit 3d39a93 into FluxML:master Jul 17, 2023
26 of 27 checks passed
@christiangnrd christiangnrd deleted the 0.14_support branch July 17, 2023 00:13
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.

None yet

3 participants