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

feat: more nested AD rules #1151

Merged
merged 5 commits into from
Jan 1, 2025
Merged

feat: more nested AD rules #1151

merged 5 commits into from
Jan 1, 2025

Conversation

avik-pal
Copy link
Member

@avik-pal avik-pal commented Dec 31, 2024

fixes #1150

  • logsoftmax & softmax
    • Impl
    • Tests
  • MaxPool & MeanPool
    • Impl
    • Tests

@avik-pal avik-pal linked an issue Dec 31, 2024 that may be closed by this pull request
15 tasks
Copy link
Contributor

github-actions bot commented Dec 31, 2024

Benchmark Results (ASV)

main 4dc08d1... main/4dc08d1eed50bf...
basics/overhead 0.129 ± 0.0015 μs 0.132 ± 0.0012 μs 0.979
time_to_load 0.89 ± 0.0094 s 0.89 ± 0.0068 s 1

Benchmark Plots

A plot of the benchmark results have been uploaded as an artifact to the workflow run for this PR.
Go to "Actions"->"Benchmark a pull request"->[the most recent run]->"Artifacts" (at the bottom).

@avik-pal avik-pal merged commit 63d3434 into main Jan 1, 2025
32 of 58 checks passed
@avik-pal avik-pal deleted the ap/nested_ad branch January 1, 2025 00:00
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.

CUDA 2nd order AD with MaxPool and logsoftmax
1 participant