bugfix

test/test_max.py

@@ -38,6 +38,7 @@ def test_scatter_max(str):
 
 
 @pytest.mark.parametrize('str', tensor_strs)
+@pytest.mark.skipif(not torch.cuda.is_available(), reason='no CUDA')
 def test_scatter_cuda_max(str):
     input = [[2, 0, 1, 4, 3], [0, 2, 1, 3, 4]]
     index = [[4, 5, 4, 2, 3], [0, 0, 2, 2, 1]]

torch_scatter/functions/utils.py

@@ -12,7 +12,7 @@ def gen_filled_tensor(input, size, fill_value):
 def gen_output(index, input, dim, dim_size, fill_value):
     if dim_size is None:
         dim_size = index.max() + 1
-        dim_size = dim.size if torch.is_tensor(input) else dim_size.data[0]
+        dim_size = dim_size if torch.is_tensor(input) else dim_size.data[0]
 
     size = list(index.size())
     size[dim] = dim_size