1a8ca5d5d6ca5cb0e3494cd5c9e77759909ec955,torch_geometric/nn/functional/max_pool_voxel/max_pool_voxel_test.py,MaxPoolVoxelGPUTest,test_max_pool_voxel,#MaxPoolVoxelGPUTest#,12

Before Change

        row = [0, 1, 2, 2, 2, 2, 3, 3, 3, 4, 5, 5]
        col = [2, 3, 0, 3, 4, 5, 1, 2, 5, 2, 2, 3]
        index = torch.cuda.LongTensor([row, col])
        weight = torch.cuda.ones(12)
        adj = torch.cuda.sparse.FloatTensor(index, weight, torch.Size([6, 6]))
        position = [[-2, -2], [2, 2], [-1, -1], [1, 1], [-2, 2], [2, -2]]
        position = torch.cuda.FloatTensor(position)

After Change

        row = [0, 1, 2, 2, 2, 2, 3, 3, 3, 4, 5, 5]
        col = [2, 3, 0, 3, 4, 5, 1, 2, 5, 2, 2, 3]
        index = torch.cuda.LongTensor([row, col])
        weight = torch.ones(12).cuda()
        adj = torch.cuda.sparse.FloatTensor(index, weight, torch.Size([6, 6]))
        position = [[-2, -2], [2, 2], [-1, -1], [1, 1], [-2, 2], [2, -2]]
        position = torch.cuda.FloatTensor(position)

Instances