added some documentation

killah-t-cell · Nov 24, 2021 · 0bdf402 · 0bdf402
1 parent 15ffaca
commit 0bdf402
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/solve.jl b/src/solve.jl
@@ -43,6 +43,10 @@ function solve(plasma::CollisionlessPlasma;
         error("lower bound must be larger than upper bound")
     end
 
+    if GPU && strategy == QuadratureTraining()
+        error("QuadratureTraining does not have GPU support. Use another strategy (e.g. StochasticTraining(200)) instead")
+    end
+
     # constants
     dim = 3
     geometry = plasma.geometry.f # this might change with a geometry refactor
@@ -157,6 +161,10 @@ function solve(plasma::ElectrostaticPlasma;
         error("lower bound must be larger than upper bound")
     end
 
+    if GPU && strategy == QuadratureTraining()
+        error("QuadratureTraining does not have GPU support. Use another strategy (e.g. StochasticTraining(200)) instead")
+    end
+
     # constants
     geometry = plasma.geometry.f # this might change with a geometry refactor
     dis = plasma.distributions