[NVSHMEM] Enable nvshmem memory allocation

apache / tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators

https://tvm.apache.org/

Apache License 2.0

11.67k stars 3.45k forks source link

Closed cyx-6 closed 4 days ago

cyx-6 commented 1 week ago

This PR add the support of nvshmem memory allocation, and integrates it into disco.