prints

Author: rusty1s

test/test_forward.py

@@ -118,6 +118,8 @@ def test_forward(test, dtype, device):
     src = tensor(test['src'], dtype, device)
     index = tensor(test['index'], torch.long, device)
     expected = tensor(test['expected'], dtype, device)
+    print(src)
+    print(index)
 
     op = getattr(torch_scatter, 'scatter_{}'.format(test['name']))
     out = op(src, index, test['dim'], fill_value=test['fill_value'])

torch_scatter/mul.py

@@ -7,6 +7,7 @@
 class ScatterMul(Function):
     @staticmethod
     def forward(ctx, out, src, index, dim):
+        print("DRIN")
         func = get_func('scatter_mul', src)
         func(src, index, out, dim)