lucasdelimanogueira
|
38a860eba1
|
fix send to other devices and compile on cloud
|
2024-05-29 17:48:37 -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 |
|
lucasdelimanogueira
|
a846d3aa3e
|
Small fix for PiPy
|
2024-05-11 01:45:59 -03:00 |
|
lucasdelimanogueira
|
d4751341bd
|
scalar mul and sum tensor
|
2024-04-30 13:22:27 -03:00 |
|
lucasdelimanogueira
|
c40951eb6f
|
fix cuda compile separete files
|
2024-04-29 19:21:10 -03:00 |
|
lucasdelimanogueira
|
5d765ed3e0
|
Fix makefile cuda
|
2024-04-29 13:39:00 -03:00 |
|
lucasdelimanogueira
|
35c028dbeb
|
Makefile cuda
|
2024-04-29 12:50:48 -03:00 |
|
lucasdelimanogueira
|
83b7b89e7f
|
makefile cuda
|
2024-04-29 12:35:57 -03:00 |
|
lucasdelimanogueira
|
62f44fb772
|
pow and add_tensor_cuda
|
2024-04-29 02:20:31 -03:00 |
|
lucasdelimanogueira
|
48f8c1d83d
|
Add tensors (still need to fix)
|
2024-04-26 18:38:07 -03:00 |
|
lucasdelimanogueira
|
06b82066ae
|
Create and print a tensor
|
2024-04-26 00:33:28 -03:00 |
|