def __init__(self, config, index):
super(Domain, self).__init__()
self.config = config
self.index = index
self.state = {}
domain_config = self.config['domains'][index]
self.name = domain_config['name']
self.id = domain_config['id']
logging.debug('Create domain %s', self.name)
self.connection = self.connect(
domain_config.get('host', '127.0.0.1'),
domain_config.get('port', 8932)
)
self.server = RPC(self.connection)
self.broker = RPCBrokerProxy(self.connection, 'broker_proxy',
self.id)
self.domain = RPCBrokerProxy(self.connection, 'broker_domain_proxy',
self.id, self.index)
def run_tests(agent):
run_test_proxy(agent)
agent.connect_server(agent.config['host'], agent.config['port'])
run_test_numpy_input(agent)
domain_id = uuid.UUID('39684e0d-6173-4d41-8efe-add8f24dd2c1')
domain = RPCBrokerProxy(
agent.server_socket, 'broker_proxy',
domain_id
)
try:
agent.server.domain_start(id=domain_id)
ensure_domain_state(agent, domain_id, 'blank')
except RPCException:
pass
assert domain.ping.wait() == 'pong'
assert domain.ping() is None
assert domain.ping.wait() == 'pong'
assert domain.ping.no_reply() is None
agent.server.domain_stop(id=domain_id)
D1 = uuid.UUID('39684e0d-6173-4d41-8efe-add8f24dd2c1')
D2 = uuid.UUID('39684e0d-6173-4d41-8efe-add8f24dd2c2')
D3 = uuid.UUID('39684e0d-6173-4d41-8efe-add8f24dd2c3')
remote_domain1 = RPCBrokerProxy(
agent.server_socket, 'broker_domain_proxy',
D1,
domain_index=0
)
remote_domain2 = RPCBrokerProxy(
agent.server_socket, 'broker_domain_proxy',
D2,
domain_index=1
)
remote_domain3 = RPCBrokerProxy(
agent.server_socket, 'broker_domain_proxy',
D3,
domain_index=2
)
config = {
'layers': [
{
'name': 'V1',
'threshold': 20000,
'relaxation': 1000,
'width': 20,
'height': 20,
'spike_cost': 11,
'connect': [
{
'name': 'V2',
'radius': 1,
'shift': [0, 0],
},
],
},
{
'name': 'V2',
'threshold': 10000,
'relaxation': 2000,
'width': 10,
'height': 10,
'is_inhibitory': True,
},
],
'domains': [
{
'name' : 'D1',
'id': D1,
'device' : {
'type': 'RandomTest',
},
'layers' : [
{'name': 'V1', 'shape': [0, 0, 10, 10]},
{'name': 'V1', 'shape': [10, 0, 10, 10]},
],
},
{
'name' : 'D2',
'id': D2,
'device' : {
'type': 'RandomTest',
},
'layers' : [
{'name': 'V1', 'shape': [10, 10, 10, 10]},
{'name': 'V1', 'shape': [0, 10, 10, 10]},
],
},
{
'name' : 'D3',
'id': D3,
'device' : {
'type': 'OpenCL',
},
'layers' : [
{'name': 'V2'},
],
},
],
}
net = None
try:
net = Net(config)
net.create()
net.upload_config()
net.deploy_domains()
net.deploy_layers()
net.deploy_neurons()
net.pre_deploy_synapses()
net.deploy_synapses()
net.post_deploy_synapses()
net.post_deploy()
time.sleep(1)
net.run()
time.sleep(4)
net.domains[0].stop()
net.domains[1].stop()
time.sleep(4)
net.stop()
time.sleep(1)
status1 = agent.server.domain_state(id=D1)
status2 = agent.server.domain_state(id=D2)
status3 = agent.server.domain_state(id=D3)
d1_stats = remote_domain1.stat.wait()
d2_stats = remote_domain2.stat.wait()
d3_stats = remote_domain3.stat.wait()
for status in [status1, status2, status3]:
try:
assert status['status'] == 'done'
except AssertionError:
print status
raise
try:
assert 'spikes_received' in d3_stats
assert d1_stats['spikes_sent'] + d2_stats['spikes_sent'] \
== d3_stats['spikes_received']
assert d1_stats['spikes_sent_to']['D3'] \
+ d2_stats['spikes_sent_to']['D3'] \
== d3_stats['spikes_received']
except (AssertionError, KeyError):
for row in [d1_stats, d2_stats, d3_stats]:
print row
raise
finally:
if net:
net.destroy()
net.clean()
class Domain(Transport):
"""
Содержит в себе настройки для конкретного домена
"""
def __init__(self, config, index):
super(Domain, self).__init__()
self.config = config
self.index = index
self.state = {}
domain_config = self.config['domains'][index]
self.name = domain_config['name']
self.id = domain_config['id']
logging.debug('Create domain %s', self.name)
self.connection = self.connect(
domain_config.get('host', '127.0.0.1'),
domain_config.get('port', 8932)
)
self.server = RPC(self.connection)
self.broker = RPCBrokerProxy(self.connection, 'broker_proxy',
self.id)
self.domain = RPCBrokerProxy(self.connection, 'broker_domain_proxy',
self.id, self.index)
def refresh_state(self):
"""
Обновляет информацию о состоянии удаленного домена
"""
self.state = self.server.domain_state(id=self.id)
def create(self):
"""
Посылает команду серверу на создание пустого домена (без нейронов и
синапсов)
"""
logging.debug('Create remote domain %s', self.name)
if not self.server.domain_start(name=self.name, id=self.id, wait=False):
raise DomainError('Domain "%s" creation failed' % self.name)
def upload_config(self):
"""
Загружает конфиг
"""
logging.debug('Upload config to remote domain %s', self.name)
self.broker.config(self.config)
def deploy_domains(self):
"""
Создает удаленные домены указывая какие из них будут локальными, а какие
глобальными
"""
logging.debug('%s.deploy_domains()', self.name)
self.broker.deploy_domains([self.name])
def __getattr__(self, name):
return SimpleRCPCall(self.broker, name, self.name)
def destroy(self):
"""
Посылает серверу команду на завершение работы домена.
"""
try:
self.server.domain_stop(id=self.id)
except RPCException as error:
logging.warning(str(error.result.get('error')))
def clean(self):
"""
Закрывает соединение.
"""
logging.debug('Clean domain %s', self.name)
self.clean_sockets()
