More conservative memory allocation.

Signed-off by: David Rowenhorst <[email protected]>
drowenhorst-nrl · Jun 6, 2024 · 323a802 · 323a802
1 parent 3a73165
commit 323a802
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyebsdindex/nlpar.py b/pyebsdindex/nlpar.py
@@ -35,7 +35,7 @@
 
 
 if _ray_installed and _pyopencl_installed:
-  from pyebsdindex.opencl.nlpar_cl import NLPAR
+  from pyebsdindex.opencl.nlpar_clray import NLPAR
 elif _pyopencl_installed and not _ray_installed:
   from pyebsdindex.opencl.nlpar_cl import NLPAR
 else:

diff --git a/pyebsdindex/opencl/nlpar_clray.py b/pyebsdindex/opencl/nlpar_clray.py
@@ -479,7 +479,7 @@ def calcnlpar_clray(self, searchradius=None, lam = None, dthresh = None, saturat
                                          rescale = rescale,
                                          gpu_id= gpu_id)
 
-    target_mem = clparams.gpu[gpu_id].max_mem_alloc_size//3
+    target_mem = clparams.gpu[gpu_id].max_mem_alloc_size//4
     max_mem = clparams.gpu[gpu_id].global_mem_size*0.4
     if target_mem*ngpuwrker > max_mem:
       target_mem = max_mem/ngpuwrker