From 6f750b6aa2d6afa9f8f837974578dfa5de5a5fd0 Mon Sep 17 00:00:00 2001
From: Yicheng Wu <yicheng.wu@ucalgary.ca>
Date: Sat, 25 Nov 2023 17:10:13 +0800
Subject: [PATCH] Update SophonLuxCUDAExt.jl

---
 ext/SophonLuxCUDAExt.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ext/SophonLuxCUDAExt.jl b/ext/SophonLuxCUDAExt.jl
index 6705bea6..23cbd78d 100644
--- a/ext/SophonLuxCUDAExt.jl
+++ b/ext/SophonLuxCUDAExt.jl
@@ -1,6 +1,6 @@
 module SophonLuxCUDAExt
 
-using Lux, LuxCUDA, Sophon, Optimization
+using Lux, LuxCUDA, Sophon, Optimization, Adapt
 
 function (::LuxCUDADevice)(prob::OptimizationProblem)
     u0 = adapt(CuArray, prob.u0)