Skip to content

Commit

Permalink
adding ChainRulesCore to CUDA extension
Browse files Browse the repository at this point in the history
  • Loading branch information
@leios
leios committed Jul 2, 2024
1 parent ff3dc86 commit fc30514
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions ext/MollyCUDAExt.jl
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ module MollyCUDAExt

using Molly
using CUDA
using ChainRulesCore

# CUDA specific calls for Molly
@non_differentiable CUDA.zeros(args...)
Expand Down

0 comments on commit fc30514

Please sign in to comment.