class ParallelizationConfiguration(IConfigurable): num_java_threads = NodeThreadPool.getNumJavaThreads() num_sim_GPU = NEFGPUInterface.getRequestedNumDevices() use_GPU_for_creation = WeightedCostApproximator.getUseGPU() p_num_java_threads = PInt('Number of Java Threads', num_java_threads, 1, NodeThreadPool.getMaxNumJavaThreads()) p_num_sim_GPU = PGpuCount('Number of GPU\'s for Simulation') p_use_GPU_for_creation = PGpuUse('Use GPU for Ensemble Creation') properties = [p_num_java_threads, p_num_sim_GPU, p_use_GPU_for_creation] def __init__(self): self.button = make_button('parallelization', self.do_configure, 'Configure Parallelization') self.button.enabled = True def do_configure(self, event): self.p_num_java_threads.setDefaultValue( NodeThreadPool.getNumJavaThreads()) self.p_num_sim_GPU.setDefaultValue( NEFGPUInterface.getRequestedNumDevices()) self.p_use_GPU_for_creation.setDefaultValue( WeightedCostApproximator.getUseGPU()) uc = ca.nengo.ui.configurable.managers.UserTemplateConfigurer(self) try: uc.configureAndWait() except ConfigException, e: e.defaultHandleBehavior()
def do_configure(self, event): self.p_num_java_threads.setDefaultValue(NodeThreadPool.getNumJavaThreads()) self.p_num_sim_GPU.setDefaultValue(NEFGPUInterface.getRequestedNumDevices()) self.p_use_GPU_for_creation.setDefaultValue(WeightedCostApproximator.getUseGPU()) uc=ca.nengo.ui.configurable.managers.UserTemplateConfigurer(self) try: uc.configureAndWait() except ConfigException, e: e.defaultHandleBehavior()
def __init__(self,network): self.nodes=[] self.projections=[] self.tasks=[] self.network=network self.initialize(network) if NodeThreadPool.getNumJavaThreads() > 0: NEFGPUInterface.requireAllOutputsOnCPU(True); self.thread_pool=NodeThreadPool(network,[]); else: self.thread_pool=None;
def do_configure(self, event): self.p_num_java_threads.setDefaultValue( NodeThreadPool.getNumJavaThreads()) self.p_num_sim_GPU.setDefaultValue( NEFGPUInterface.getRequestedNumDevices()) self.p_use_GPU_for_creation.setDefaultValue( WeightedCostApproximator.getUseGPU()) uc = ca.nengo.ui.configurable.managers.UserTemplateConfigurer(self) try: uc.configureAndWait() except ConfigException, e: e.defaultHandleBehavior()