Skip to content

Commit

Permalink
Make executor threadpool size configurable
Browse files Browse the repository at this point in the history
fix #94
  • Loading branch information
drelu committed Jul 15, 2013
1 parent 91c2a54 commit 242cdb3
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 1 deletion.
6 changes: 5 additions & 1 deletion bigjob/bigjob_agent.py
Expand Up @@ -97,12 +97,15 @@ def __init__(self, args):
# linked under mpirun_rsh # linked under mpirun_rsh
if default_dict.has_key("mpirun"): if default_dict.has_key("mpirun"):
self.MPIRUN=default_dict["mpirun"] self.MPIRUN=default_dict["mpirun"]

if default_dict.has_key("number_executor_threads"):
THREAD_POOL_SIZE=int(default_dict["number_executor_threads"])

self.OUTPUT_TAR=False self.OUTPUT_TAR=False
if default_dict.has_key("create_output_tar"): if default_dict.has_key("create_output_tar"):
self.OUTPUT_TAR=eval(default_dict["create_output_tar"]) self.OUTPUT_TAR=eval(default_dict["create_output_tar"])
logger.debug("Create output tar: %r", self.OUTPUT_TAR) logger.debug("Create output tar: %r", self.OUTPUT_TAR)



self.failed_polls = 0 self.failed_polls = 0


############################################################################## ##############################################################################
Expand Down Expand Up @@ -190,6 +193,7 @@ def __init__(self, args):


############################################################################## ##############################################################################
# start background thread for polling new jobs and monitoring current jobs # start background thread for polling new jobs and monitoring current jobs
logger.debug("Creating executor thread pool of size: %d"%(THREAD_POOL_SIZE))
self.resource_lock=threading.RLock() self.resource_lock=threading.RLock()
self.threadpool = ThreadPool(THREAD_POOL_SIZE) self.threadpool = ThreadPool(THREAD_POOL_SIZE)


Expand Down
3 changes: 3 additions & 0 deletions bigjob_agent.conf
Expand Up @@ -5,6 +5,9 @@ cpr=False
shell = /bin/bash shell = /bin/bash
mpirun = mpirun mpirun = mpirun


# control multi-threaded compute unit execution
number_executor_threads=3

# Lauch Method # Lauch Method
# Default launch method is ssh # Default launch method is ssh
# Future support for aprun (e.g. for Kraken) # Future support for aprun (e.g. for Kraken)
Expand Down

0 comments on commit 242cdb3

Please sign in to comment.