thanks for the great library, I can still hardly believe the amazing performance.
When executing this script, I get an compilation error in the backwards pass through the sumsoftmaxweight
reduction.
loss: 6528.2978515625
Compiling libKeOpstorch509fe71999 in /home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37//build-libKeOpstorch509fe71999:
formula: Grad_WithSavedForward(Max_SumShiftExpWeight_Reduction(Sum((Var(0,10,0) - Var(1,10,1))),1,Concat(IntCst(1),(Var(0,10,0) - Var(1,10,1)))), Var(0,10,0), Var(2,12,1), Var(3,12,1))
aliases: Var(0,10,0); Var(1,10,1); Var(2,12,1); Var(3,12,1);
dtype : float32
... /home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/formulas/maths/Extract.h(23): error: static assertion failed with "Index out of bound in Extract"
detected during:
instantiation of class "keops::Extract<F, START, DIM_> [with F=keops::Extract<keops::Var<2, 12, 1>, 1, 11>, START=1, DIM_=11]"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/formulas/maths/Subtract.h(134): here
instantiation of class "keops::Subtract_Alias<FA, keops::Zero<DIM>> [with FA=keops::IdOrZero<keops::Var<0, 10, 0>, keops::Var<0, 10, 0>, keops::Extract<keops::Extract<keops::Var<2, 12, 1>, 1, 11>, 1, 11>>, DIM=10]"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/formulas/maths/Subtract.h(23): here
instantiation of type "keops::Subtract<keops::IdOrZero<keops::Var<0, 10, 0>, keops::Var<0, 10, 0>, keops::Extract<keops::Extract<keops::Var<2, 12, 1>, 1, 11>, 1, 11>>, keops::IdOrZero<keops::Var<1, 10, 1>, keops::Var<0, 10, 0>, keops::Extract<keops::Extract<keops::Var<2, 12, 1>, 1, 11>, 1, 11>>>"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/formulas/maths/Subtract.h(48): here
instantiation of type "keops::Subtract_Impl<FA, FB>::DiffT<keops::Var<0, 10, 0>, keops::Extract<keops::Extract<keops::Var<2, 12, 1>, 1, 11>, 1, 11>> [with FA=keops::Var<0, 10, 0>, FB=keops::Var<1, 10, 1>]"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/formulas/maths/Concat.h(37): here
instantiation of type "keops::Concat_Impl<F, G>::DiffTG<keops::Var<0, 10, 0>, keops::Extract<keops::Extract<keops::Var<2, 12, 1>, 1, 11>, 1, 11>> [with F=keops::IntConstant<1>, G=keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>]"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/formulas/maths/Concat.h(40): here
[ 2 instantiation contexts not shown ]
instantiation of type "keops::Grad<keops::Scal<keops::Exp<keops::Subtract<keops::Sum<keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>, keops::Extract<keops::Var<3, 12, 1>, 0, 1>>>, keops::Concat<keops::IntConstant<1>, keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>>, keops::Var<0, 10, 0>, keops::Extract<keops::Var<2, 12, 1>, 1, 11>>"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/reductions/Sum_Reduction.h(72): here
instantiation of type "keops::Sum_Reduction_Impl<F, tagI>::DiffT<keops::Var<0, 10, 0>, keops::Extract<keops::Var<2, 12, 1>, 1, 11>, void> [with F=keops::Scal<keops::Exp<keops::Subtract<keops::Sum<keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>, keops::Extract<keops::Var<3, 12, 1>, 0, 1>>>, keops::Concat<keops::IntConstant<1>, keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>>, tagI=1]"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/autodiff/Grad.h(16): here
instantiation of type "keops::Grad<keops::Sum_Reduction<keops::Scal<keops::Exp<keops::Subtract<keops::Sum<keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>, keops::Extract<keops::Var<3, 12, 1>, 0, 1>>>, keops::Concat<keops::IntConstant<1>, keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>>, 1>, keops::Var<0, 10, 0>, keops::Extract<keops::Var<2, 12, 1>, 1, 11>>"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/reductions/Max_SumShiftExp_Reduction.h(114): here
instantiation of type "keops::Max_SumShiftExp_Reduction<F, tagI, G_>::DiffT<keops::Var<0, 10, 0>, keops::Var<2, 12, 1>, keops::Var<3, 12, 1>> [with F=keops::Sum<keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>, tagI=1, G_=keops::Concat<keops::IntConstant<1>, keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>]"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/autodiff/Grad.h(20): here
instantiation of type "keops::Grad_WithSavedForward<keops::Max_SumShiftExp_Reduction<keops::Sum<keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>, 1, keops::Concat<keops::IntConstant<1>, keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>>, keops::Var<0, 10, 0>, keops::Var<2, 12, 1>, keops::Var<3, 12, 1>>"
/home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/libKeOpstorch509fe71999.h(21): here
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/formulas/maths/Subtract.h(134): error: static assertion failed with "Dimensions must be the same for Subtract"
detected during:
instantiation of class "keops::Subtract_Alias<FA, keops::Zero<DIM>> [with FA=keops::IdOrZero<keops::Var<0, 10, 0>, keops::Var<0, 10, 0>, keops::Extract<keops::Extract<keops::Var<2, 12, 1>, 1, 11>, 1, 11>>, DIM=10]"
(23): here
instantiation of type "keops::Subtract<keops::IdOrZero<keops::Var<0, 10, 0>, keops::Var<0, 10, 0>, keops::Extract<keops::Extract<keops::Var<2, 12, 1>, 1, 11>, 1, 11>>, keops::IdOrZero<keops::Var<1, 10, 1>, keops::Var<0, 10, 0>, keops::Extract<keops::Extract<keops::Var<2, 12, 1>, 1, 11>, 1, 11>>>"
(48): here
instantiation of type "keops::Subtract_Impl<FA, FB>::DiffT<keops::Var<0, 10, 0>, keops::Extract<keops::Extract<keops::Var<2, 12, 1>, 1, 11>, 1, 11>> [with FA=keops::Var<0, 10, 0>, FB=keops::Var<1, 10, 1>]"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/formulas/maths/Concat.h(37): here
instantiation of type "keops::Concat_Impl<F, G>::DiffTG<keops::Var<0, 10, 0>, keops::Extract<keops::Extract<keops::Var<2, 12, 1>, 1, 11>, 1, 11>> [with F=keops::IntConstant<1>, G=keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>]"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/formulas/maths/Concat.h(40): here
instantiation of type "keops::Concat_Impl<F, G>::DiffT<keops::Var<0, 10, 0>, keops::Extract<keops::Var<2, 12, 1>, 1, 11>> [with F=keops::IntConstant<1>, G=keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>]"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/formulas/maths/Scal.h(59): here
instantiation of type "keops::Scal_Impl<FA, FB>::DiffT<keops::Var<0, 10, 0>, keops::Extract<keops::Var<2, 12, 1>, 1, 11>> [with FA=keops::Exp<keops::Subtract<keops::Sum<keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>, keops::Extract<keops::Var<3, 12, 1>, 0, 1>>>, FB=keops::Concat<keops::IntConstant<1>, keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>]"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/autodiff/Grad.h(16): here
instantiation of type "keops::Grad<keops::Scal<keops::Exp<keops::Subtract<keops::Sum<keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>, keops::Extract<keops::Var<3, 12, 1>, 0, 1>>>, keops::Concat<keops::IntConstant<1>, keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>>, keops::Var<0, 10, 0>, keops::Extract<keops::Var<2, 12, 1>, 1, 11>>"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/reductions/Sum_Reduction.h(72): here
instantiation of type "keops::Sum_Reduction_Impl<F, tagI>::DiffT<keops::Var<0, 10, 0>, keops::Extract<keops::Var<2, 12, 1>, 1, 11>, void> [with F=keops::Scal<keops::Exp<keops::Subtract<keops::Sum<keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>, keops::Extract<keops::Var<3, 12, 1>, 0, 1>>>, keops::Concat<keops::IntConstant<1>, keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>>, tagI=1]"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/autodiff/Grad.h(16): here
instantiation of type "keops::Grad<keops::Sum_Reduction<keops::Scal<keops::Exp<keops::Subtract<keops::Sum<keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>, keops::Extract<keops::Var<3, 12, 1>, 0, 1>>>, keops::Concat<keops::IntConstant<1>, keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>>, 1>, keops::Var<0, 10, 0>, keops::Extract<keops::Var<2, 12, 1>, 1, 11>>"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/reductions/Max_SumShiftExp_Reduction.h(114): here
instantiation of type "keops::Max_SumShiftExp_Reduction<F, tagI, G_>::DiffT<keops::Var<0, 10, 0>, keops::Var<2, 12, 1>, keops::Var<3, 12, 1>> [with F=keops::Sum<keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>, tagI=1, G_=keops::Concat<keops::IntConstant<1>, keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>]"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/autodiff/Grad.h(20): here
instantiation of type "keops::Grad_WithSavedForward<keops::Max_SumShiftExp_Reduction<keops::Sum<keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>, 1, keops::Concat<keops::IntConstant<1>, keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>>, keops::Var<0, 10, 0>, keops::Var<2, 12, 1>, keops::Var<3, 12, 1>>"
/home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/libKeOpstorch509fe71999.h(21): here
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/formulas/maths/Add.h(152): error: static assertion failed with "Dimensions must be the same for Add"
detected during:
instantiation of class "keops::Add_Alias<keops::Zero<DIM>, FB> [with FB=keops::Subtract<keops::IdOrZero<keops::Var<0, 10, 0>, keops::Var<0, 10, 0>, keops::Extract<keops::Extract<keops::Var<2, 12, 1>, 1, 11>, 1, 11>>, keops::IdOrZero<keops::Var<1, 10, 1>, keops::Var<0, 10, 0>, keops::Extract<keops::Extract<keops::Var<2, 12, 1>, 1, 11>, 1, 11>>>, DIM=10]"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/formulas/norms/Scalprod.h(33): here
instantiation of type "keops::Add<keops::Zero<10>, keops::Subtract<keops::IdOrZero<keops::Var<0, 10, 0>, keops::Var<0, 10, 0>, keops::Extract<keops::Extract<keops::Var<2, 12, 1>, 1, 11>, 1, 11>>, keops::IdOrZero<keops::Var<1, 10, 1>, keops::Var<0, 10, 0>, keops::Extract<keops::Extract<keops::Var<2, 12, 1>, 1, 11>, 1, 11>>>>"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/formulas/maths/Concat.h(40): here
instantiation of type "keops::Concat_Impl<F, G>::DiffT<keops::Var<0, 10, 0>, keops::Extract<keops::Var<2, 12, 1>, 1, 11>> [with F=keops::IntConstant<1>, G=keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>]"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/formulas/maths/Scal.h(59): here
instantiation of type "keops::Scal_Impl<FA, FB>::DiffT<keops::Var<0, 10, 0>, keops::Extract<keops::Var<2, 12, 1>, 1, 11>> [with FA=keops::Exp<keops::Subtract<keops::Sum<keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>, keops::Extract<keops::Var<3, 12, 1>, 0, 1>>>, FB=keops::Concat<keops::IntConstant<1>, keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>]"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/autodiff/Grad.h(16): here
instantiation of type "keops::Grad<keops::Scal<keops::Exp<keops::Subtract<keops::Sum<keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>, keops::Extract<keops::Var<3, 12, 1>, 0, 1>>>, keops::Concat<keops::IntConstant<1>, keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>>, keops::Var<0, 10, 0>, keops::Extract<keops::Var<2, 12, 1>, 1, 11>>"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/reductions/Sum_Reduction.h(72): here
instantiation of type "keops::Sum_Reduction_Impl<F, tagI>::DiffT<keops::Var<0, 10, 0>, keops::Extract<keops::Var<2, 12, 1>, 1, 11>, void> [with F=keops::Scal<keops::Exp<keops::Subtract<keops::Sum<keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>, keops::Extract<keops::Var<3, 12, 1>, 0, 1>>>, keops::Concat<keops::IntConstant<1>, keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>>, tagI=1]"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/autodiff/Grad.h(16): here
instantiation of type "keops::Grad<keops::Sum_Reduction<keops::Scal<keops::Exp<keops::Subtract<keops::Sum<keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>, keops::Extract<keops::Var<3, 12, 1>, 0, 1>>>, keops::Concat<keops::IntConstant<1>, keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>>, 1>, keops::Var<0, 10, 0>, keops::Extract<keops::Var<2, 12, 1>, 1, 11>>"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/reductions/Max_SumShiftExp_Reduction.h(114): here
instantiation of type "keops::Max_SumShiftExp_Reduction<F, tagI, G_>::DiffT<keops::Var<0, 10, 0>, keops::Var<2, 12, 1>, keops::Var<3, 12, 1>> [with F=keops::Sum<keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>, tagI=1, G_=keops::Concat<keops::IntConstant<1>, keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>]"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/autodiff/Grad.h(20): here
instantiation of type "keops::Grad_WithSavedForward<keops::Max_SumShiftExp_Reduction<keops::Sum<keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>, 1, keops::Concat<keops::IntConstant<1>, keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>>, keops::Var<0, 10, 0>, keops::Var<2, 12, 1>, keops::Var<3, 12, 1>>"
/home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/libKeOpstorch509fe71999.h(21): here
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/formulas/maths/Add.h(41): error: static assertion failed with "Dimensions must be the same for Add"
detected during:
instantiation of class "keops::Add_Impl<FA, FB> [with FA=keops::Subtract<keops::IdOrZero<keops::Var<0, 10, 0>, keops::Var<0, 10, 0>, keops::SumT<keops::Mult<keops::Exp<keops::Subtract<keops::Sum<keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>, keops::Extract<keops::Var<3, 12, 1>, 0, 1>>>, keops::Scalprod<keops::Extract<keops::Var<2, 12, 1>, 1, 11>, keops::Concat<keops::IntConstant<1>, keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>>>, 10>>, keops::IdOrZero<keops::Var<1, 10, 1>, keops::Var<0, 10, 0>, keops::SumT<keops::Mult<keops::Exp<keops::Subtract<keops::Sum<keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>, keops::Extract<keops::Var<3, 12, 1>, 0, 1>>>, keops::Scalprod<keops::Extract<keops::Var<2, 12, 1>, 1, 11>, keops::Concat<keops::IntConstant<1>, keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>>>, 10>>>, FB=keops::Scal<keops::Exp<keops::Subtract<keops::Sum<keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>, keops::Extract<keops::Var<3, 12, 1>, 0, 1>>>, keops::Add<keops::Zero<10>, keops::Subtract<keops::IdOrZero<keops::Var<0, 10, 0>, keops::Var<0, 10, 0>, keops::Extract<keops::Extract<keops::Var<2, 12, 1>, 1, 11>, 1, 11>>, keops::IdOrZero<keops::Var<1, 10, 1>, keops::Var<0, 10, 0>, keops::Extract<keops::Extract<keops::Var<2, 12, 1>, 1, 11>, 1, 11>>>>>]"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/reductions/Reduction.h(26): here
instantiation of class "keops::Reduction<F_, tagI_> [with F_=keops::Grad<keops::Scal<keops::Exp<keops::Subtract<keops::Sum<keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>, keops::Extract<keops::Var<3, 12, 1>, 0, 1>>>, keops::Concat<keops::IntConstant<1>, keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>>, keops::Var<0, 10, 0>, keops::Extract<keops::Var<2, 12, 1>, 1, 11>>, tagI_=0]"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/reductions/Sum_Reduction.h(22): here
instantiation of class "keops::Sum_Reduction_Impl<F, tagI> [with F=keops::Grad<keops::Scal<keops::Exp<keops::Subtract<keops::Sum<keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>, keops::Extract<keops::Var<3, 12, 1>, 0, 1>>>, keops::Concat<keops::IntConstant<1>, keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>>, keops::Var<0, 10, 0>, keops::Extract<keops::Var<2, 12, 1>, 1, 11>>, tagI=0]"
/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/pre_headers.h(40): here
instantiation of class "keops::KeopsNS<F> [with F=keops::Grad_WithSavedForward<keops::Max_SumShiftExp_Reduction<keops::Sum<keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>, 1, keops::Concat<keops::IntConstant<1>, keops::Subtract_Impl<keops::Var<0, 10, 0>, keops::Var<1, 10, 1>>>>, keops::Var<0, 10, 0>, keops::Var<2, 12, 1>, keops::Var<3, 12, 1>>]"
/home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/libKeOpstorch509fe71999.h(21): here
4 errors detected in the compilation of "/tmp/tmpxft_00003e29_00000000-6_link_autodiff.cpp1.ii".
CMake Error at keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o.Release.cmake:279 (message):
Error generating file
/home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/./keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o
make[3]: *** [CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o] Error 1
make[2]: *** [CMakeFiles/keopslibKeOpstorch509fe71999.dir/all] Error 2
make[1]: *** [CMakeFiles/libKeOpstorch509fe71999.dir/rule] Error 2
make: *** [libKeOpstorch509fe71999] Error 2
-- The CXX compiler identification is GNU 7.4.0
-- Check for working CXX compiler: /usr/bin/c++
-- Check for working CXX compiler: /usr/bin/c++ -- works
-- Detecting CXX compiler ABI info
-- Detecting CXX compiler ABI info - done
-- Detecting CXX compile features
-- Detecting CXX compile features - done
-- Looking for C++ include pthread.h
-- Looking for C++ include pthread.h - found
-- Looking for pthread_create
-- Looking for pthread_create - not found
-- Looking for pthread_create in pthreads
-- Looking for pthread_create in pthreads - not found
-- Looking for pthread_create in pthread
-- Looking for pthread_create in pthread - found
-- Found Threads: TRUE
-- Compute properties automatically set to: -DMAXIDGPU=6;-DMAXTHREADSPERBLOCK0=1024;-DSHAREDMEMPERBLOCK0=49152;-DMAXTHREADSPERBLOCK1=1024;-DSHAREDMEMPERBLOCK1=49152;-DMAXTHREADSPERBLOCK2=1024;-DSHAREDMEMPERBLOCK2=49152;-DMAXTHREADSPERBLOCK3=1024;-DSHAREDMEMPERBLOCK3=49152;-DMAXTHREADSPERBLOCK4=1024;-DSHAREDMEMPERBLOCK4=49152;-DMAXTHREADSPERBLOCK5=1024;-DSHAREDMEMPERBLOCK5=49152;-DMAXTHREADSPERBLOCK6=1024;-DSHAREDMEMPERBLOCK6=49152
-- The CUDA compiler identification is NVIDIA 10.0.130
-- Check for working CUDA compiler: /usr/local/cuda-10.0/bin/nvcc
-- Check for working CUDA compiler: /usr/local/cuda-10.0/bin/nvcc -- works
-- Detecting CUDA compiler ABI info
-- Detecting CUDA compiler ABI info - done
-- The CUDA Host CXX Compiler: /usr/bin/c++
-- Autodetected CUDA architecture(s): 6.1 6.1 6.1 6.1 6.1 6.1 6.1
-- Using shared_obj_name: libKeOpstorch509fe71999
-- Found PythonInterp: /home_sdc/rremme_tmp/anaconda3/envs/main/bin/python3.7 (found version "3.7.4")
-- Found PythonLibs: /home_sdc/rremme_tmp/anaconda3/envs/main/lib/libpython3.7m.so
-- pybind11 v2.3.dev1
-- Performing Test HAS_FLTO
-- Performing Test HAS_FLTO - Success
-- LTO enabled
-- Configuring done
-- Generating done
-- Build files have been written to: /home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999
--------------------- MAKE DEBUG -----------------
Command '['cmake', '--build', '.', '--target', 'libKeOpstorch509fe71999', '--', 'VERBOSE=1']' returned non-zero exit status 2.
/usr/bin/cmake -H/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops -B/home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999 --check-build-system CMakeFiles/Makefile.cmake 0
/usr/bin/make -f CMakeFiles/Makefile2 libKeOpstorch509fe71999
make[1]: Entering directory '/home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999'
/usr/bin/cmake -H/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops -B/home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999 --check-build-system CMakeFiles/Makefile.cmake 0
/usr/bin/cmake -E cmake_progress_start /home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles 5
/usr/bin/make -f CMakeFiles/Makefile2 CMakeFiles/libKeOpstorch509fe71999.dir/all
make[2]: Entering directory '/home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999'
/usr/bin/make -f CMakeFiles/keopslibKeOpstorch509fe71999.dir/build.make CMakeFiles/keopslibKeOpstorch509fe71999.dir/depend
make[3]: Entering directory '/home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999'
[ 20%] Building NVCC (Device) object CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o
cd /home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core && /usr/bin/cmake -E make_directory /home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/.
cd /home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core && /usr/bin/cmake -D verbose:BOOL=1 -D build_configuration:STRING=Release -D generated_file:STRING=/home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/./keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o -D generated_cubin_file:STRING=/home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/./keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o.cubin.txt -P /home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o.Release.cmake
-- Removing /home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/./keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o
/usr/bin/cmake -E remove /home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/./keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o
-- Generating dependency file: /home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o.NVCC-depend
/usr/local/cuda-10.0/bin/nvcc -M -D__CUDACC__ /home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/link_autodiff.cu -o /home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o.NVCC-depend -m64 -DkeopslibKeOpstorch509fe71999_EXPORTS -DMAXIDGPU=6 -DMAXTHREADSPERBLOCK0=1024 -DSHAREDMEMPERBLOCK0=49152 -DMAXTHREADSPERBLOCK1=1024 -DSHAREDMEMPERBLOCK1=49152 -DMAXTHREADSPERBLOCK2=1024 -DSHAREDMEMPERBLOCK2=49152 -DMAXTHREADSPERBLOCK3=1024 -DSHAREDMEMPERBLOCK3=49152 -DMAXTHREADSPERBLOCK4=1024 -DSHAREDMEMPERBLOCK4=49152 -DMAXTHREADSPERBLOCK5=1024 -DSHAREDMEMPERBLOCK5=49152 -DMAXTHREADSPERBLOCK6=1024 -DSHAREDMEMPERBLOCK6=49152 -D_FORCE_INLINES -DCUDA_BLOCK_SIZE=192 -DUSE_CUDA=1 -D__TYPE__=float -DC_CONTIGUOUS=1 -DMODULE_NAME=libKeOpstorch509fe71999 -D_GLIBCXX_USE_CXX11_ABI=0 -DUSE_DOUBLE=0 -DKERNEL_GEOM_TYPE=0 -DKERNEL_SIG_TYPE=0 -DKERNEL_SPHERE_TYPE=0 -DMODULE_NAME_FSHAPE_SCP=fshape_scp_gaussiangaussiangaussian_unoriented_float -Xcompiler ,\"-Wall\",\"-Wno-unknown-pragmas\",\"-fmax-errors=2\",\"-fPIC\",\"-O3\",\"-DNDEBUG\",\"-O3\" -gencode arch=compute_61,code=sm_61 --use_fast_math --compiler-options=-fPIC -ccbin /usr/bin/c++ --pre-include=libKeOpstorch509fe71999.h -DNVCC -I/usr/local/cuda-10.0/include -I/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops -I/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops -I/home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999 -I/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/torch/include -I/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/torch/include/torch/csrc/api/include
-- Generating temporary cmake readable file: /home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o.depend.tmp
/usr/bin/cmake -D input_file:FILEPATH=/home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o.NVCC-depend -D output_file:FILEPATH=/home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o.depend.tmp -D verbose=1 -P /usr/share/cmake-3.10/Modules/FindCUDA/make2cmake.cmake
-- Copy if different /home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o.depend.tmp to /home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o.depend
/usr/bin/cmake -E copy_if_different /home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o.depend.tmp /home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o.depend
-- Removing /home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o.depend.tmp and /home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o.NVCC-depend
/usr/bin/cmake -E remove /home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o.depend.tmp /home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o.NVCC-depend
-- Generating /home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/./keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o
/usr/local/cuda-10.0/bin/nvcc /home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops/core/link_autodiff.cu -c -o /home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/./keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o -m64 -DkeopslibKeOpstorch509fe71999_EXPORTS -DMAXIDGPU=6 -DMAXTHREADSPERBLOCK0=1024 -DSHAREDMEMPERBLOCK0=49152 -DMAXTHREADSPERBLOCK1=1024 -DSHAREDMEMPERBLOCK1=49152 -DMAXTHREADSPERBLOCK2=1024 -DSHAREDMEMPERBLOCK2=49152 -DMAXTHREADSPERBLOCK3=1024 -DSHAREDMEMPERBLOCK3=49152 -DMAXTHREADSPERBLOCK4=1024 -DSHAREDMEMPERBLOCK4=49152 -DMAXTHREADSPERBLOCK5=1024 -DSHAREDMEMPERBLOCK5=49152 -DMAXTHREADSPERBLOCK6=1024 -DSHAREDMEMPERBLOCK6=49152 -D_FORCE_INLINES -DCUDA_BLOCK_SIZE=192 -DUSE_CUDA=1 -D__TYPE__=float -DC_CONTIGUOUS=1 -DMODULE_NAME=libKeOpstorch509fe71999 -D_GLIBCXX_USE_CXX11_ABI=0 -DUSE_DOUBLE=0 -DKERNEL_GEOM_TYPE=0 -DKERNEL_SIG_TYPE=0 -DKERNEL_SPHERE_TYPE=0 -DMODULE_NAME_FSHAPE_SCP=fshape_scp_gaussiangaussiangaussian_unoriented_float -Xcompiler ,\"-Wall\",\"-Wno-unknown-pragmas\",\"-fmax-errors=2\",\"-fPIC\",\"-O3\",\"-DNDEBUG\",\"-O3\" -gencode arch=compute_61,code=sm_61 --use_fast_math --compiler-options=-fPIC -ccbin /usr/bin/c++ --pre-include=libKeOpstorch509fe71999.h -DNVCC -I/usr/local/cuda-10.0/include -I/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops -I/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/keops -I/home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999 -I/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/torch/include -I/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/torch/include/torch/csrc/api/include
-- Removing /home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/./keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o
/usr/bin/cmake -E remove /home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999/CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/./keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o
CMakeFiles/keopslibKeOpstorch509fe71999.dir/build.make:63: recipe for target 'CMakeFiles/keopslibKeOpstorch509fe71999.dir/keops/core/keopslibKeOpstorch509fe71999_generated_link_autodiff.cu.o' failed
make[3]: Leaving directory '/home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999'
CMakeFiles/Makefile2:104: recipe for target 'CMakeFiles/keopslibKeOpstorch509fe71999.dir/all' failed
make[2]: Leaving directory '/home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999'
CMakeFiles/Makefile2:79: recipe for target 'CMakeFiles/libKeOpstorch509fe71999.dir/rule' failed
make[1]: Leaving directory '/home_sdc/rremme_tmp/.cache/pykeops-1.2-cpython-37/build-libKeOpstorch509fe71999'
Makefile:118: recipe for target 'libKeOpstorch509fe71999' failed
--------------------- ----------- -----------------
Done.
Traceback (most recent call last):
File "sumsoftmaxweight_bug.py", line 17, in <module>
loss.backward()
File "/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/torch/tensor.py", line 118, in backward
torch.autograd.backward(self, gradient, retain_graph, create_graph)
File "/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/torch/autograd/__init__.py", line 93, in backward
allow_unreachable=True) # allow_unreachable flag
File "/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/torch/autograd/function.py", line 77, in apply
return self._forward_cls.backward(self, *args)
File "/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/torch/generic/generic_red.py", line 123, in backward
grad = genconv(formula_g, aliases_g, backend, dtype, device_id, ranges, *args_g)
File "/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/torch/generic/generic_red.py", line 21, in forward
['-DPYTORCH_INCLUDE_DIR=' + ';'.join(include_dirs)]).import_module()
File "/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/site-packages/pykeops/common/keops_io.py", line 52, in import_module
return importlib.import_module(self.dll_name)
File "/home_sdc/rremme_tmp/anaconda3/envs/main/lib/python3.7/importlib/__init__.py", line 127, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
File "<frozen importlib._bootstrap>", line 1006, in _gcd_import
File "<frozen importlib._bootstrap>", line 983, in _find_and_load
File "<frozen importlib._bootstrap>", line 965, in _find_and_load_unlocked
ModuleNotFoundError: No module named 'libKeOpstorch509fe71999'