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

Merge NNlibCUDA as subpackage #471

Merged
merged 168 commits into from
Feb 14, 2023
Merged

Merge NNlibCUDA as subpackage #471

merged 168 commits into from
Feb 14, 2023

Conversation

ToucheSir
Copy link
Member

As part of the process of turning NNlibCUDA into a package extension, we need to first move the source into this repo while preserving its history. Without doing so, we wouldn't be able to re-register it as a subpackage. To that end, I've followed the procedure described in JuliaPlots/Plots.jl#4419 (comment) to place it under ext/NNlibCUDA. Once this is done and registered, #445 can be redone on top of this foundation.

PR Checklist

  • [N/A] Tests are added
  • [N/A] Documentation, if applicable

ToucheSir and others added 25 commits June 18, 2022 16:56
save allocs during algorithm search
* print/convert batchedadjtrans over cuarray

* Update test/batchedadjtrans.jl

Co-authored-by: Brian Chen <[email protected]>

Co-authored-by: Brian Chen <[email protected]>
* move ctc loss from Flux

* fixup

* trivial

* rm cpu
Missing Identity Activation Import
1.7 nightly no longer exists and was causing errors. Also updated to 1.8 to match configuration across the org.
added fold/unfold and gpu tests
* add dropout

* bump
* overload NNlib._rng_compat_array

* trigger CI
agents:
queue: "juliagpu"
cuda: "*"
timeout_in_minutes: 60
Copy link
Member

Choose a reason for hiding this comment

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

not blocking for this PR, cleanup can be done later, but subpackages' .buildkite and .github dirs should be removed?

Copy link
Member Author

Choose a reason for hiding this comment

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

Yes, absolutely.

@ToucheSir ToucheSir merged commit 09347ea into master Feb 14, 2023
@ToucheSir ToucheSir deleted the nnlibcuda-merge branch February 18, 2023 16:15
@ToucheSir ToucheSir mentioned this pull request Feb 18, 2023
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.