From 790348b47213d21c589955d63581747535a63e20 Mon Sep 17 00:00:00 2001
From: Tiago Freitas Pereira <tiagofrepereira@gmail.com>
Date: Tue, 10 May 2016 17:50:08 +0200
Subject: [PATCH] Taking care of the issue #19
---
bob/bio/base/config/grid/gpu.py | 16 ++++++++++++++++
bob/bio/base/grid.py | 3 ++-
setup.py | 1 +
3 files changed, 19 insertions(+), 1 deletion(-)
create mode 100644 bob/bio/base/config/grid/gpu.py
diff --git a/bob/bio/base/config/grid/gpu.py b/bob/bio/base/config/grid/gpu.py
new file mode 100644
index 00000000..ecc6e475
--- /dev/null
+++ b/bob/bio/base/config/grid/gpu.py
@@ -0,0 +1,16 @@
+import bob.bio.base
+
+# define a queue with demanding parameters
+grid = bob.bio.base.grid.Grid(
+ training_queue = 'GPU',
+ # preprocessing
+ preprocessing_queue = '4G',
+ # feature extraction
+ extraction_queue = 'GPU',
+ # feature projection
+ projection_queue = '4G',
+ # model enrollment
+ enrollment_queue = '4G',
+ # scoring
+ scoring_queue = '4G'
+)
diff --git a/bob/bio/base/grid.py b/bob/bio/base/grid.py
index f11fbd3b..447575de 100644
--- a/bob/bio/base/grid.py
+++ b/bob/bio/base/grid.py
@@ -29,7 +29,8 @@ PREDEFINED_QUEUES = {
'16G-io-big' : {'queue' : 'q1dm', 'memfree' : '16G', 'pe_opt' : 'pe_mth 2', 'hvmem' : '8G', 'io_big' : True},
'32G' : {'queue' : 'q1dm', 'memfree' : '32G', 'pe_opt' : 'pe_mth 4', 'hvmem' : '8G', 'io_big' : True},
'64G' : {'queue' : 'q1dm', 'memfree' : '64G', 'pe_opt' : 'pe_mth 8', 'hvmem' : '8G', 'io_big' : True},
- 'Week' : {'queue' : 'q1wm', 'memfree' : '32G', 'pe_opt' : 'pe_mth 4', 'hvmem' : '8G'}
+ 'Week' : {'queue' : 'q1wm', 'memfree' : '32G', 'pe_opt' : 'pe_mth 4', 'hvmem' : '8G'},
+ 'GPU' : {'queue' : 'gpu'}
}
class Grid:
diff --git a/setup.py b/setup.py
index 25e288b4..5d0650d7 100644
--- a/setup.py
+++ b/setup.py
@@ -145,6 +145,7 @@ setup(
'local-p16 = bob.bio.base.config.grid.local:grid_p16',
'grid = bob.bio.base.config.grid.grid:grid',
'demanding = bob.bio.base.config.grid.demanding:grid',
+ 'gpu = bob.bio.base.config.grid.gpu:grid',
],
},
--
GitLab