Suggestion Description
The signature of put and get should be changed to copy. Copy will have the signature:
@triton.jit
def copy(src_ptr, dst_ptr, from_rank, to_rank, heap_bases, mask=None):
Operating System
No response
GPU
No response
ROCm Component
No response