Commit graph

6 commits

Author SHA1 Message Date
lucasdelimanogueira
0e9c4779a8 add argument src to broadcast distributed 2024-05-25 09:38:14 -03:00
lucasdelimanogueira
1e0c408ffb fix import distributed nn 2024-05-25 08:57:35 -03:00
lucasdelimanogueira
29ee569d04 fix import distributed 2024-05-24 19:55:19 -03:00
lucasdelimanogueira
94b3582a46 test broadcast and all reduce 2024-05-24 18:31:59 -03:00
lucasdelimanogueira
55db13836b broadcast allreduce nccl 2024-05-24 18:19:39 -03:00
lucasdelimanogueira
e55c1195ec implemented mpi for distributed run 2024-05-24 15:12:53 -03:00