From fc30514943d1b4f8b4f7f7fb3d6cea40a73f60c2 Mon Sep 17 00:00:00 2001 From: James Schloss Date: Tue, 2 Jul 2024 12:04:10 +0200 Subject: [PATCH] adding ChainRulesCore to CUDA extension --- ext/MollyCUDAExt.jl | 1 + 1 file changed, 1 insertion(+) diff --git a/ext/MollyCUDAExt.jl b/ext/MollyCUDAExt.jl index 5a14a479..a6a0b9c8 100644 --- a/ext/MollyCUDAExt.jl +++ b/ext/MollyCUDAExt.jl @@ -2,6 +2,7 @@ module MollyCUDAExt using Molly using CUDA +using ChainRulesCore # CUDA specific calls for Molly @non_differentiable CUDA.zeros(args...)