Binyang Li adc9ee5684 Export mscclpp GpuBuffer to dlpack format (#492) ...

For mscclpp, to use nvls we require the buffer is allocated by
mscclpp::GpuBuffer. Due to cupy doesn't support bfloat16 yet, we export
the raw buffer to dlpack format.
User can use this feature to create buffer with type supported by
pytorch
```python
buffer = RawGpuBuffer(1024 * 2) # 2 for bfloat16
dl_pack = buffer.to_dlpack(str(torch.bfloat16))
tensor = torch.utils.dlpack.from_dlpack(dl_pack)
```

2025-04-03 12:59:32 -07:00

..

ISSUE_TEMPLATE

Add a documentation issue template (#230)

2023-12-05 01:01:45 +00:00

workflows

Export mscclpp GpuBuffer to dlpack format (#492)

2025-04-03 12:59:32 -07:00