Skip to content
This repository was archived by the owner on Jul 26, 2025. It is now read-only.

Control point Enzyme gradients with CUDABackend#96

Draft
SouthEndMusic wants to merge 2 commits into
masterfrom
gpu_gradient
Draft

Control point Enzyme gradients with CUDABackend#96
SouthEndMusic wants to merge 2 commits into
masterfrom
gpu_gradient

Conversation

@SouthEndMusic
Copy link
Copy Markdown
Owner

@SouthEndMusic SouthEndMusic commented Feb 4, 2025

The custom reverse rule seems to have made the code stable, but (pre)compile time is extremely long

To do:

  • Add FiniteDiff.jl comparison test for the loss function

@SouthEndMusic SouthEndMusic marked this pull request as draft February 4, 2025 20:19
@codecov-commenter
Copy link
Copy Markdown

Codecov Report

Attention: Patch coverage is 20.00000% with 16 lines in your changes missing coverage. Please review.

Project coverage is 91.71%. Comparing base (802171d) to head (ed2f496).
Report is 2 commits behind head on master.

Files with missing lines Patch % Lines
ext/SplineGridsEnzymeExt.jl 18.75% 13 Missing ⚠️
src/spline_grid.jl 25.00% 3 Missing ⚠️
Additional details and impacted files
@@            Coverage Diff             @@
##           master      #96      +/-   ##
==========================================
- Coverage   92.94%   91.71%   -1.24%     
==========================================
  Files          15       15              
  Lines        1191     1207      +16     
==========================================
  Hits         1107     1107              
- Misses         84      100      +16     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants