Fix indexing for AMD GPUs

Due to https://github.com/cupy/cupy/issues/7676 BlockIndexing did not work correctly on AMD GPUs. This is MR fixes it.