I meant using your trick

I think the difference is hard to solve

👍

Cc @Sören Brunk

@Fabio Grätz do you think you can update the PR?

or should i create a new PR. What I want is an easy capability of training on a single node with multiple gpus

Got it, first single node multi GPU 👍 I can implement this on Friday or Saturday and then ping you for review. Or do you need it before that?

@Fabio Grätz also maybe we should try to integrate accelerate in the same way - Hopefully @Evan Sadler can help 😄

Also maybe - https://www.deepspeed.ai/getting-started/

@Fabio Grätz have you used this ^

I haven’t used it much but ~1.5 years ago “I got an example to train with it” on k8s (which they don’t explicitly mentioned as supported in the docs). Ultimately under the hood it also just uses

torch.distributed.init_process_group()

, see here. Back then I just created a kubeflow PytorchJob to run it which worked. Image needed

nvidia-cuda-toolkit

. To summarize, at the state of ~1.5 years ago I think it would already have been supported.

Still draft PRs because I will add more tests and docs: • https://github.com/flyteorg/flytekit/pull/1583 • https://github.com/flyteorg/flyteplugins/pull/343 • https://github.com/flyteorg/flyteidl/pull/394 But torch elastic task now works for me when executing locally, with

nnodes=1

in a single pod, and with

nnodes>1

with the pytorch operator. I think we could try with alpaca now 🦙 The problems with rendezvous flakiness I mentioned in the call on Thursday were actually related to network config on my notebook (no ipv6 enabled).

[W socket.cpp:601] [c10d] The IPv6 network addresses of (1.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.ip6.arpa, 49651) cannot be retrieved (gai error: 8 - nodename nor servname provided, or not known).

I have one question about the

execute

method I copied from

PythonFunctionTask

: We don’t need the else case here for dynamic even though the original docstring hints one should implement it as well, right?

cc @Niels Bantilan did you end up trying alpaca?

@Fabio Grätz what do you think should we merge the idl PR?

how can i help take them over the finish line?

I will add tests and documentation on the weekend. Then I’ll request PR reviews You could help by testing it, so far I have only run minimal working examples (e.g. this one) that don’t do much more other than making sure that the process group can be initialized.

the code works, successfully ran the workflow on the

facebook/opt-125m

, currently trying to get to work on a pre-existing llama model on huggingface

also still need to test it on multiple cpus/gpus

@Niels Bantilan I can help with multi core example

cool, I just updated our fork/branch with my changes: https://github.com/unionai-oss/stanford_alpaca/tree/flytekit-alpaca

One other thing about which I’m interested in your opinion:

torchrun

allows the user to set

--nnodes

which could e.g. be

2

but also be

"1:2"

which means min 1 max 2. Currently this is what iour new

task_config=Elastic()

exposes as well. The kubeflow PytorchJob allows setting

minReplicas

,

maxReplicas

(which by default are both None), and

replicas

(see here). In theory you could say min 2, max 4, replicas 3 (without going into how much sense this makes). If a user specifies

2:3

we currently set min to 2 and max and replicas to 3. To summarize: Should we expose

nnodes

like torchrun or

min_replicas

,

max_replicas

, and

replicas

like the pytorchjob to the user?

Ready for review from my side:

• https://github.com/flyteorg/flyte/issues/3614 • https://github.com/flyteorg/flytekit/pull/1603 • https://github.com/flyteorg/flyteidl/pull/394 • https://github.com/flyteorg/flyteplugins/pull/343 • https://github.com/flyteorg/flytesnacks/pull/987

How does the merge process typically look like when idl is changed? Tests in flytekit and flyteplugins fail since idl changes are not there yet

cc +@Byron Hsu

@Byron Hsu we are enabling torch-elastic in flytekit now

@Fabio Grätz / @Byron Hsu seems these instructions are no longer valid - https://docs.flyte.org/en/latest/deployment/plugins/k8s/index.html#deployment-plugin-setup-k8s - as we have one training operator now. cc @Kevin Su / @Yuvraj

Also, @Fabio Grätz do you folks use - https://github.com/libffcv/ffcv?

@Niels Bantilan / @James Sutton / @Evan Sadler

Thanks for finishing the Pr and merging 🚀