### Suggestion Description The signature of `put` and `get` should be changed to `copy`. Copy will have the signature: ```python @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_