TODO: faire fonctionner le code sur gpu... pb dans les indices dans le kernel ?