fux default jastrow on gpu

NLESC-JCER · Dec 11, 2023 · bc50100 · bc50100
1 parent 5c89bc7
commit bc50100
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/qmctorch/wavefunction/slater_jastrow.py b/qmctorch/wavefunction/slater_jastrow.py
@@ -194,7 +194,9 @@ def init_jastrow(self, jastrow):
 
             # create a simple Pade Jastrow factor as default
             if jastrow == 'default':
-                self.jastrow = JastrowFactorElectronElectron(self.mol, PadeJastrowKernel)
+                self.jastrow = JastrowFactorElectronElectron(self.mol, 
+                                                             PadeJastrowKernel, 
+                                                             cuda=self.cuda)
 
             elif isinstance(jastrow, list):
                 self.jastrow = CombineJastrow(jastrow)