geoprior.utils.parallel_utils#

Parallel execution helpers for GeoPrior workflows.

Functions

apply_gpu_env(env, *, gpu_id[, allow_growth])

apply_tf_threading(*, intra, inter)

apply_thread_env(env, *, n_jobs[, threads, ...])

detect_gpu_ids(*[, env])

pick_gpu_id(idx, gpu_ids)

resolve_device(device, *[, env])

resolve_gpu_ids(gpu_ids, *[, env])

resolve_n_jobs(n_jobs)

threads_per_job(*, n_jobs[, threads, reserve])

geoprior.utils.parallel_utils.resolve_n_jobs(n_jobs)[source]#
Parameters:

n_jobs (int)

Return type:

int

geoprior.utils.parallel_utils.threads_per_job(*, n_jobs, threads=0, reserve=1)[source]#
Parameters:
Return type:

int

geoprior.utils.parallel_utils.apply_thread_env(env, *, n_jobs, threads=0, reserve=1)[source]#
Parameters:
Return type:

dict[str, str]

geoprior.utils.parallel_utils.apply_tf_threading(*, intra, inter)[source]#
Parameters:
Return type:

None

geoprior.utils.parallel_utils.detect_gpu_ids(*, env=None)[source]#
Parameters:

env (dict[str, str] | None)

Return type:

list[str]

geoprior.utils.parallel_utils.resolve_device(device, *, env=None)[source]#
Parameters:
Return type:

str

geoprior.utils.parallel_utils.resolve_gpu_ids(gpu_ids, *, env=None)[source]#
Parameters:
Return type:

list[str]

geoprior.utils.parallel_utils.pick_gpu_id(idx, gpu_ids)[source]#
Parameters:
Return type:

str | None

geoprior.utils.parallel_utils.apply_gpu_env(env, *, gpu_id, allow_growth=True)[source]#
Parameters:
Return type:

dict[str, str]