LeErnst:

As an input it receives indices (tuple of LongTensor) and i am wondering whether this function is supported on the GPU or does it move all the involved data to the RAM and use the CPU. I think a tuple is a python object and therefore not available on the GPU, isn’t it?


Yes, index_put is available on the GPU and the data from the passed Python types is extracted.
The CUDA implementation calls index_put)impl_ and dispatches to index_put_with_sort_stub if accumulate=True or deterministic algorithms are selected, which then calls index_put_with_sort_kernel.