Skip to content

Commit fc30514

Browse files
committed
adding ChainRulesCore to CUDA extension
1 parent ff3dc86 commit fc30514

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

ext/MollyCUDAExt.jl

+1
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@ module MollyCUDAExt
22

33
using Molly
44
using CUDA
5+
using ChainRulesCore
56

67
# CUDA specific calls for Molly
78
@non_differentiable CUDA.zeros(args...)

0 commit comments

Comments
 (0)