MCPcopy Index your code
hub / github.com/apache/tvm / on_data

Method on_data

python/tvm/rpc/proxy.py:103–120  ·  view source on GitHub ↗

on data

(self, message)

Source from the content-addressed store, hash-verified

101 self._proxy.handler_ready(self)
102
103 def on_data(self, message):
104 """on data"""
105 assert isinstance(message, bytes)
106 if self.forward_proxy:
107 self.forward_proxy.send_data(message)
108 else:
109 while message and self._init_req_nbytes > len(self._init_message):
110 nbytes = self._init_req_nbytes - len(self._init_message)
111 self._init_message += message[:nbytes]
112 message = message[nbytes:]
113 if self._init_req_nbytes == len(self._init_message):
114 temp = self._init_message
115 self._init_req_nbytes = 0
116 self._init_message = b""
117 self._init_step(temp)
118 if message:
119 logging.info("Invalid RPC protocol, too many bytes %s", self.name())
120 self.close()
121
122 def on_error(self, err):
123 logging.info("%s: Error in RPC %s", self.name(), err)

Callers 2

on_messageMethod · 0.80
on_messageMethod · 0.80

Calls 5

_init_stepMethod · 0.95
nameMethod · 0.95
infoMethod · 0.80
closeMethod · 0.65
send_dataMethod · 0.45

Tested by

no test coverage detected