https://flyte.org logo
#contribute
Title
# contribute
b

Byron Hsu

05/09/2023, 5:20 AM
Allow 0 worker in pytorch plugins & Add objectMeta to PyTorchJob: https://github.com/flyteorg/flyteplugins/pull/348 Enable pod template and Use copy to construct head/worker in ray plugin: https://github.com/flyteorg/flyteplugins/pull/349 PR to fix pytorch and ray plugin @Fabio Grätz @Kevin Su
d

Dan Rammer (hamersaw)

05/09/2023, 3:18 PM
left a few small comments, looks great @Byron Hsu!
k

Ketan (kumare3)

05/10/2023, 4:31 AM
updated the comment
that being said you can leave the code, but i think we should make flytekit, just send a single container task?
b

Byron Hsu

05/10/2023, 4:32 AM
runs without needing an operator and does not need a CRD to be created.
this makes sense
ok will raise a pr to flytekit
k

Ketan (kumare3)

05/10/2023, 4:46 AM
nice
thank you
highly appreciated
f

Fabio Grätz

05/10/2023, 7:35 AM
torch.distributed.init_process_group()
requires some env vars to be set (world size, rank, …). If we use a single container task we need to overwrite
execute
and set these before calling
super().execute
?