|
layers.py
|
Add CUDA graph-based all reduce launcher (#26)
|
2023-04-05 11:16:57 -07:00 |
|
mappings.py
|
Support tensor parallel (#2)
|
2023-03-21 13:45:42 -07:00 |
|
random.py
|
Support tensor parallel (#2)
|
2023-03-21 13:45:42 -07:00 |
|
utils.py
|
Support tensor parallel (#2)
|
2023-03-21 13:45:42 -07:00 |