typo

1f915586 · rusty1s · 65abb5e3 · 1f915586
Commit 1f915586 authored Mar 16, 2018 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

torch_spline_conv/functions/ffi.py torch_spline_conv/functions/ffi.py +3 -0

No files found.
--- a/torch_spline_conv/functions/ffi.py
+++ b/torch_spline_conv/functions/ffi.py
@@ -45,9 +45,12 @@ def spline_weighting_backward_input(grad_output, weight, basis,
                                    weight_index):  # pragma: no cover
    grad_input = grad_output.new(grad_output.size(0), weight.size(1))
    func = get_func('weighting_backward_input', grad_output)
+
+    # Transpose for coalesced memory access.
    weight = weight.transpose(1, 2).contiguous()
    func(grad_input, grad_output, weight, basis, weight_index)
    weight = weight.transpose(1, 2).contiguous()
+
    return grad_input