From 1f795a1525c86f477ac0ea24bbde3922d9f11c26 Mon Sep 17 00:00:00 2001
From: Zuheng <zuhengdavid@gmail.com>
Date: Tue, 15 Aug 2023 12:07:38 -0700
Subject: [PATCH] @require cuda ext

---
 src/NormalizingFlows.jl | 3 +++
 test/runtests.jl        | 1 +
 2 files changed, 4 insertions(+)

diff --git a/src/NormalizingFlows.jl b/src/NormalizingFlows.jl
index 2d70892..7b7c2b9 100644
--- a/src/NormalizingFlows.jl
+++ b/src/NormalizingFlows.jl
@@ -94,6 +94,9 @@ function __init__()
         @require Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f" include(
             "../ext/NormalizingFlowsZygoteExt.jl"
         )
+        @require CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba" include(
+            "../ext/NormalizingFlowsCUDAExt.jl"
+        )
     end
 end
 end
diff --git a/test/runtests.jl b/test/runtests.jl
index e050a64..574b762 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -7,6 +7,7 @@ using ADTypes, DiffResults
 using ForwardDiff, Zygote, Enzyme, ReverseDiff
 using Test
 
+include("cuda.jl")
 include("ad.jl")
 include("objectives.jl")
 include("interface.jl")
\ No newline at end of file