Enable GPU support in trajectory bundle problem solver using CUDA

aarontrowbridge · Feb 17, 2025 · c677bc1 · c677bc1
1 parent 07b31be
commit c677bc1
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -61,6 +61,7 @@ using LinearAlgebra
 using SparseArrays
 using NamedTrajectories
 using TrajectoryBundles
+using CUDA
 
 # construct dynamical generators: elements of real isomorphism of the Lie algebra 𝔰𝔲(2)
 
@@ -197,6 +198,8 @@ prob = TrajectoryBundleProblem(traj, f, r_loss, rs, cs;
 TrajectoryBundles.solve!(prob;
     max_iter = 200,
     σ₀ = 1.0e0,
+    gpu = true,
+    gpu_backend = CUDA.CUDABackend()
 )
 
 # plot best trajectory