@require cuda ext

TuringLang · Aug 15, 2023 · 1f795a1 · 1f795a1
1 parent 0dde183
commit 1f795a1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git 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
@@ -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")