Skip to content
Snippets Groups Projects
Commit d079c8c5 authored by Andrew Cohen's avatar Andrew Cohen
Browse files

init USE_CUDA in Segment.getImages for spmd use

parent d7a9d3bb
Branches
No related tags found
No related merge requests found
......@@ -55,6 +55,7 @@ end
% write the NLM settings (for cache validation)
h5create(targetFile,'/Metadata/NLM',4);
h5write(targetFile,'/Metadata/NLM',segParams.NLM);
4;
function prepareH5(targetFile,im)
szim=size(im);
......
......@@ -35,6 +35,11 @@ if false==segParams.denoise
end
global USE_CUDA
if isempty(USE_CUDA)
% when we run batchdenoise, in the spmd this global won't be set
% make sure to set it here...
USE_CUDA=segParams.useCuda&&~ismac && HIP.Cuda.DeviceCount>0;
end
bUseCuda=USE_CUDA;
if 4~=length(segParams.NLM)
bUseCuda=false;
......
No preview for this file type
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment