Changeset 915 for trunk

Timestamp:

01/27/11 11:57:12 (2 years ago)

Author:

apdavison

Message:

Discovered that when using NEST with mpi4py, you must import nest first and let it do the MPI initialization. The only time this seems to be a problem with PyNN is if a user imports pyNN.random before pyNN.nest. It would be nice to handle this more gracefully, but for now I've just added a test that NEST and mpi4py agree on the rank, and a hopefully useful error message.

Location:

trunk

Files:

• examples/IF_curr_exp2.py (modified) (1 diff)
• examples/VAbenchmarks.py (modified) (2 diffs)
• examples/VAbenchmarks2-csa.py (modified) (2 diffs)
• examples/VAbenchmarks2.py (modified) (2 diffs)
• examples/VAbenchmarks3.py (modified) (2 diffs)
• src/nest/__init__.py (modified) (1 diff)

trunk/examples/IF_curr_exp2.py

@@ -13 +13 @@
 
 import numpy
-from pyNN.random import NumpyRNG
 from pyNN.utility import get_script_args
-
 simulator_name = get_script_args(1)[0]  
 exec("from pyNN.%s import *" % simulator_name)
-
+from pyNN.random import NumpyRNG
 
 setup(timestep=0.01, min_delay=2.0, max_delay=4.0)

trunk/examples/VAbenchmarks.py

@@ -20 +20 @@
 from math import *
 
-from pyNN.random import NumpyRNG, RandomDistribution
 from pyNN.utility import get_script_args, Timer
 usage = """Usage: python VAbenchmarks.py <simulator> <benchmark>
@@ -27 +26 @@
 simulator_name, benchmark = get_script_args(2, usage)  
 exec("from pyNN.%s import *" % simulator_name)
+from pyNN.random import NumpyRNG, RandomDistribution
 
 timer = Timer()

trunk/examples/VAbenchmarks2-csa.py

@@ -24 +24 @@
 import csa
 
-from pyNN.random import NumpyRNG, RandomDistribution
 from pyNN.utility import get_script_args, Timer
 usage = """Usage: python VAbenchmarks.py <simulator> <benchmark>
@@ -31 +30 @@
 simulator_name, benchmark = get_script_args(2, usage)  
 exec("from pyNN.%s import *" % simulator_name)
+from pyNN.random import NumpyRNG, RandomDistribution
 
 timer = Timer()

trunk/examples/VAbenchmarks2.py

@@ -23 +23 @@
 from math import *
 
-from pyNN.random import NumpyRNG, RandomDistribution
 from pyNN.utility import get_script_args, Timer
 usage = """Usage: python VAbenchmarks.py <simulator> <benchmark>
@@ -30 +29 @@
 simulator_name, benchmark = get_script_args(2, usage)  
 exec("from pyNN.%s import *" % simulator_name)
+from pyNN.random import NumpyRNG, RandomDistribution
 
 timer = Timer()

trunk/examples/VAbenchmarks3.py

@@ -23 +23 @@
 from math import *
 
-from pyNN.random import NumpyRNG, RandomDistribution
 from pyNN.utility import get_script_args, Timer
 usage = """Usage: python VAbenchmarks.py <simulator> <benchmark>
@@ -30 +29 @@
 simulator_name, benchmark = get_script_args(2, usage)  
 exec("from pyNN.%s import *" % simulator_name)
+from pyNN.random import NumpyRNG, RandomDistribution
 
 timer = Timer()

trunk/src/nest/__init__.py

@@ -10 +10 @@
 common.simulator = simulator
 recording.simulator = simulator
+
+if nest.Rank() != recording.mpi_comm.rank:
+    raise Exception("MPI not working properly. Please make sure you import pyNN.nest before pyNN.random.")
 
 import numpy