skygpu
/
skynet
mirror of https://github.com/skygpu/skynet.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Testing unlimited msg size

pull/2/head
Guillermo Rodriguez 2022-12-24 11:28:17 -03:00
parent 8427eaafd7
commit 55781688cd
No known key found for this signature in database
GPG Key ID: EC3AB66D5D83B392
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
skynet/brain.py
View File

@ -406,8 +406,8 @@ async def run_skynet(
own_cert_string=tls_cert_data) own_cert_string=tls_cert_data)
with ( with (
pynng.Rep0() as rpc_sock, pynng.Rep0(recv_max_size=0) as rpc_sock,
pynng.Bus0() as dgpu_bus pynng.Bus0(recv_max_size=0) as dgpu_bus
): ):
async with open_database_connection( async with open_database_connection(
db_user, db_pass, db_host) as db_pool: db_user, db_pass, db_host) as db_pool:

2
skynet/dgpu.py
View File

@ -177,7 +177,7 @@ async def open_dgpu_node(
ca_string=skynet_cert_data) ca_string=skynet_cert_data)
logging.info(f'connecting to {dgpu_address}') logging.info(f'connecting to {dgpu_address}')
with pynng.Bus0() as dgpu_sock: with pynng.Bus0(recv_max_size=0) as dgpu_sock:
dgpu_sock.tls_config = tls_config dgpu_sock.tls_config = tls_config
dgpu_sock.dial(dgpu_address) dgpu_sock.dial(dgpu_address)

2
skynet/frontend/__init__.py
View File

@ -70,7 +70,7 @@ async def open_skynet_rpc(
own_cert_string=tls_cert_data, own_cert_string=tls_cert_data,
ca_string=skynet_cert_data) ca_string=skynet_cert_data)
with pynng.Req0() as sock: with pynng.Req0(recv_max_size=0) as sock:
if security: if security:
sock.tls_config = tls_config sock.tls_config = tls_config