-
Notifications
You must be signed in to change notification settings - Fork 10.5k
[AutoDiff] Replace SIMD use of @differentiable(jvp:vjp:) with @derivative(of:) #28930
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
[AutoDiff] Replace SIMD use of @differentiable(jvp:vjp:) with @derivative(of:) #28930
Conversation
ab854f4
to
88c5f65
Compare
@swift-ci Please test tensorflow |
The reason is A workaround is to mark |
Mark `SIMD._vjpInit(repeating:)` as `@inlinable`.
@swift-ci Please test tensorflow |
@swift-ci Please test tensorflow macOS |
With the change of using
@derivative(of:)
as done in this PR, I started off with some parts of TF-1085 by updating SIMD.