Skip to content
Snippets Groups Projects
Commit 7313bb39 authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Merge branch 'config-resources' into 'master'

Add dask-client configurations as resources

Closes #19

See merge request !40
parents 656b9671 1bba35ad
Branches
Tags
1 merge request!40Add dask-client configurations as resources
Pipeline #44081 passed
from dask.distributed import Client
from bob.pipelines.distributed.sge import SGEMultipleQueuesCluster
from bob.pipelines.distributed.sge_queues import QUEUE_DEMANDING
cluster = SGEMultipleQueuesCluster(min_jobs=20, sge_job_spec=QUEUE_DEMANDING)
dask_client = Client(cluster)
...@@ -101,56 +101,3 @@ QUEUE_LIGHT = { ...@@ -101,56 +101,3 @@ QUEUE_LIGHT = {
This queue setup has a light arrangement. This queue setup has a light arrangement.
For CPU jobs, it prioritizes all.q and not io_big For CPU jobs, it prioritizes all.q and not io_big
""" """
QUEUE_DEMANDING = {
"default": {
"queue": "q_1day",
"memory": "8GB",
"io_big": True,
"resource_spec": "",
"max_jobs": 48,
"resources": "",
},
"q_1day": {
"queue": "q_1day",
"memory": "4GB",
"io_big": False,
"resource_spec": "",
"max_jobs": 48,
"resources": {"q_1day": 1},
},
"q_1week": {
"queue": "q_1week",
"memory": "4GB",
"io_big": True,
"resource_spec": "",
"resources": {"q_1week": 1},
},
"q_short_gpu": {
"queue": "q_short_gpu",
"memory": "30GB",
"io_big": False,
"resource_spec": "",
"max_jobs": 45,
"resources": {"q_short_gpu": 1},
},
"q_gpu": {
"queue": "q_gpu",
"memory": "30GB",
"io_big": False,
"resource_spec": "",
"resources": {"q_gpu": 1},
},
"q_long_gpu": {
"queue": "q_long_gpu",
"memory": "30GB",
"io_big": False,
"resource_spec": "",
"resources": {"q_long_gpu": 1},
},
}
"""
This queue setup has a light arrangement.
For CPU jobs, it prioritizes all.q and not io_big
"""
...@@ -14,6 +14,7 @@ install_requires = load_requirements() ...@@ -14,6 +14,7 @@ install_requires = load_requirements()
setup( setup(
# This is the basic information about the project.
name="bob.pipelines", name="bob.pipelines",
version=open("version.txt").read().rstrip(), version=open("version.txt").read().rstrip(),
description="Tools to build robust and extensible pipelines", description="Tools to build robust and extensible pipelines",
...@@ -27,13 +28,29 @@ setup( ...@@ -27,13 +28,29 @@ setup(
# maintainer_email='email@example.com' # maintainer_email='email@example.com'
# you may add more keywords separating those by commas (a, b, c, ...) # you may add more keywords separating those by commas (a, b, c, ...)
keywords="bob", keywords="bob",
# If you have a better, long description of your package, place it on the
# 'doc' directory and then hook it here
long_description=open("README.rst").read(), long_description=open("README.rst").read(),
# leave this here, it is pretty standard
# This line is required for any distutils based packaging.
packages=find_packages(), packages=find_packages(),
include_package_data=True, include_package_data=True,
zip_safe=False, zip_safe=False,
# Packages that should be installed when you "install" this package.
install_requires=install_requires, install_requires=install_requires,
# check classifiers, add and remove as you see fit
# entry_points defines which scripts will be inside the 'bin' directory
entry_points = {
'dask.client': [
'local-parallel = bob.pipelines.config.distributed.local_parallel',
'sge = bob.pipelines.config.distributed.sge_default',
'sge-light = bob.pipelines.config.distributed.sge_light',
],
},
# check classifiers (important for PyPI), add and remove as you see fit.
# full list here: https://pypi.org/classifiers/ # full list here: https://pypi.org/classifiers/
# don't remove the Bob framework unless it's not a bob package # don't remove the Bob framework unless it's not a bob package
classifiers=[ classifiers=[
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment