@@ -209,13 +209,7 @@ async def teardown(self) -> None:
209
209
assert self .pong_mesh is not None
210
210
await self .pong_mesh .stop ()
211
211
212
- def setup_env (self ) -> None :
213
- # ensure any changes are reflected
214
- os .environ ["HYPERACTOR_DEFAULT_ENCODING" ] = "serde_bincode"
215
- os .environ ["HYPERACTOR_CHANNEL_MULTIPART" ] = "0"
216
-
217
212
async def setup (self ) -> None :
218
- self .setup_env ()
219
213
reload_config_from_env ()
220
214
pong_mesh = proc_mesh (hosts = self .host_count , gpus = self .gpu_count )
221
215
await pong_mesh .logging_option (stream_to_client = True , aggregate_window_sec = None )
@@ -225,12 +219,6 @@ async def setup(self) -> None:
225
219
self .message = bytes (self .message_size )
226
220
227
221
228
- class ActorLatencyMultipart (ActorLatency ):
229
- def setup_env (self ) -> None :
230
- os .environ ["HYPERACTOR_DEFAULT_ENCODING" ] = "serde_multipart"
231
- os .environ ["HYPERACTOR_CHANNEL_MULTIPART" ] = "1"
232
-
233
-
234
222
@dataclass
235
223
class ActorThroughput (ActorLatency ):
236
224
request_batch_size : int = 10
@@ -253,22 +241,14 @@ async def run_once(self) -> None:
253
241
self .bump_counter ("messages" , pong .size () * self .request_batch_size )
254
242
255
243
256
- class ActorThroughputMultipart (ActorThroughput ):
257
- def setup_env (self ) -> None :
258
- os .environ ["HYPERACTOR_DEFAULT_ENCODING" ] = "serde_multipart"
259
- os .environ ["HYPERACTOR_CHANNEL_MULTIPART" ] = "1"
260
-
261
-
262
- message_sizes : list [int ] = [10 ** n for n in range (1 , 9 )]
263
- host_counts = [1 , 8 , 16 ]
264
- gpu_counts = [1 , 8 ]
244
+ message_sizes : list [int ] = [10 ** n for n in range (3 , 9 , 2 )]
245
+ host_counts = [1 , 10 ]
246
+ gpu_counts = [1 , 10 ]
265
247
host_counts = [1 ]
266
248
gpu_counts = [1 ]
267
249
runners = [
268
250
ActorLatency ,
269
- ActorLatencyMultipart ,
270
251
ActorThroughput ,
271
- ActorThroughputMultipart ,
272
252
]
273
253
274
254
for hosts , gpus , message_size , Runner in itertools .product (
0 commit comments