After I run with CUDA_LAUNCH_BLOCKING=1, the indicated array which is out of index bound are actually in the index. Are you sure this is about the index out of bound error?
Also, I have a question about the the for loop variable allocation. For example,
for i in range(batch_size):
    points = input_dict['points'][i]

here, input_dict[‘points’] is a list of 4 2-d tensors, the size of tensor is different size. Will this doable in training process?