diff --git a/pyebsdindex/nlpar.py b/pyebsdindex/nlpar.py index d20c5aa..e1fc4cd 100644 --- 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 index 564129e..3a1123c 100644 --- 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