def send_control_info(control_info, destination): """Send control info to destination """ import string msg = string.join(control_info,',') send_string(msg, destination, control_tag)
def send_control_info(control_info, destination):
"""Send control info to destination
"""
import string
msg = string.join(control_info, ',')
send_string(msg, destination, control_tag)
def send(x,
destination,
use_buffer=False,
vanilla=False,
tag=default_tag,
bypass=False):
"""Wrapper for easy MPI send.
Send x to destination.
Automatically determine appropriate protocol
and call corresponding send function.
Also passes type and size information on as preceding message to
simplify the receive call.
The variable x can be any (picklable) type, but
numpy variables and text strings will most efficient.
Setting vanilla = 1 forces vanilla mode for any type.
If use_buffer is True, workspace x will be used for the return value.
In this case the corresponding receive call must specify a buffer.
Otherwise a new workspace will be created by receive.
If bypass is True, all admin and error checks
get bypassed to reduce the latency. Should only
be used for sending numpy arrays and should be matched
with a bypass in the corresponding receive command.
"""
import types, string
if bypass:
send_array(x, destination, tag)
return
# Input check
errmsg = 'Destination id (%s) must be an integer.' % destination
assert type(destination) == types.IntType, errmsg
errmsg = 'Tag %d is reserved by pypar - please use another.' % control_tag
assert tag != control_tag, errmsg
# Create metadata about object to be sent
control_info, x = create_control_info(x, vanilla, return_object=True)
protocol = control_info[0]
# Possibly transmit control data
if use_buffer is False:
send_control_info(control_info, destination)
# Transmit payload data
if protocol == 'array':
send_array(x, destination, tag)
elif protocol in ['string', 'vanilla']:
send_string(x, destination, tag)
else:
raise 'Unknown protocol: %s' % protocol
def send(x, destination, use_buffer=False, vanilla=False,
tag=default_tag, bypass=False):
"""Wrapper for easy MPI send.
Send x to destination.
Automatically determine appropriate protocol
and call corresponding send function.
Also passes type and size information on as preceding message to
simplify the receive call.
The variable x can be any (picklable) type, but
numpy variables and text strings will most efficient.
Setting vanilla = 1 forces vanilla mode for any type.
If use_buffer is True, workspace x will be used for the return value.
In this case the corresponding receive call must specify a buffer.
Otherwise a new workspace will be created by receive.
If bypass is True, all admin and error checks
get bypassed to reduce the latency. Should only
be used for sending numpy arrays and should be matched
with a bypass in the corresponding receive command.
"""
import types, string
if bypass:
send_array(x, destination, tag)
return
# Input check
errmsg = 'Destination id (%s) must be an integer.' % destination
assert type(destination) == types.IntType, errmsg
errmsg = 'Tag %d is reserved by pypar - please use another.' % control_tag
assert tag != control_tag, errmsg
# Create metadata about object to be sent
control_info, x = create_control_info(x, vanilla, return_object=True)
protocol = control_info[0]
# Possibly transmit control data
if use_buffer is False:
send_control_info(control_info, destination)
# Transmit payload data
if protocol == 'array':
send_array(x, destination, tag)
elif protocol in ['string', 'vanilla']:
send_string(x, destination, tag)
else:
raise 'Unknown protocol: %s' %protocol
def send_control_info(control_info, destination):
"""Send control info to destination
"""
import string
# Convert to strings
control_info = [str(c) for c in control_info]
control_msg = string.join(control_info, control_sep)
if len(control_msg) > control_data_max_size:
errmsg = 'Length of control_info exceeds specified maximium (%d)'\
%control_data_max_size
errmsg += ' - Please increase it (in pypar.py)'
raise errmsg
send_string(control_msg, destination, control_tag)
def send_control_info(control_info, destination):
"""Send control info to destination
"""
import string
#Convert to strings
control_info = [str(c) for c in control_info]
control_msg = string.join(control_info,control_sep)
if len(control_msg) > control_data_max_size:
errmsg = 'Length of control_info exceeds specified maximium (%d)'\
%control_data_max_size
errmsg += ' - Please increase it (in pypar.py)'
raise errmsg
send_string(control_msg, destination, control_tag)
def raw_send(x, destination, tag=0, vanilla=0):
"""Wrapper for raw MPI send.
Send x to destination with tag.
Automatically determine appropriate protocol
and call corresponding send function.
The variable x can be any (picklable) type, but
Numeric variables and text strings will most efficient.
Setting vanilla = 1 forces vanilla mode for any type.
"""
protocol = get_control_info(x, vanilla)[0]
if protocol == 'array':
send_array(x, destination, tag)
elif protocol == 'string':
send_string(x, destination, tag)
else:
send_vanilla(x, destination, tag)
def raw_send(x, destination, tag=0, vanilla=0):
"""Wrapper for raw MPI send.
Send x to destination with tag.
Automatically determine appropriate protocol
and call corresponding send function.
The variable x can be any (picklable) type, but
Numeric variables and text strings will most efficient.
Setting vanilla = 1 forces vanilla mode for any type.
"""
protocol = get_control_info(x, vanilla)[0]
if protocol == 'array':
send_array(x, destination, tag)
elif protocol == 'string':
send_string(x, destination, tag)
else:
send_vanilla(x, destination, tag)
def send(x, destination, tag=0, vanilla=0):
"""Wrapper for easy MPI send.
Send x to destination with tag.
Automatically determine appropriate protocol
and call corresponding send function.
Also passes type and size information on as preceding message to
simplify the receive call.
The variable x can be any (picklable) type, but
Numeric variables and text strings will most efficient.
Setting vanilla = 1 forces vanilla mode for any type.
"""
import string
control_info = get_control_info(x, vanilla)
protocol = control_info[0]
if protocol == 'array':
send_control_info(control_info, destination)
send_array(x, destination, tag)
elif protocol == 'string':
send_control_info(control_info, destination)
send_string(x, destination, tag)
elif protocol == 'vanilla':
from cPickle import dumps
s = dumps(x, 1)
control_info[2] = str(len(s))
send_control_info(control_info, destination)
send_string(s, destination, tag)
else:
raise "Unknown values for protocol: %s" % protocol
def send(x, destination, tag=0, vanilla=0):
"""Wrapper for easy MPI send.
Send x to destination with tag.
Automatically determine appropriate protocol
and call corresponding send function.
Also passes type and size information on as preceding message to
simplify the receive call.
The variable x can be any (picklable) type, but
Numeric variables and text strings will most efficient.
Setting vanilla = 1 forces vanilla mode for any type.
"""
import string
control_info = get_control_info(x, vanilla)
protocol = control_info[0]
if protocol == 'array':
send_control_info(control_info, destination)
send_array(x, destination, tag)
elif protocol == 'string':
send_control_info(control_info, destination)
send_string(x, destination, tag)
elif protocol == 'vanilla':
from cPickle import dumps
s = dumps(x, 1)
control_info[2] = str(len(s))
send_control_info(control_info, destination)
send_string(s, destination, tag)
else:
raise "Unknown values for protocol: %s" %protocol
