En cas de suspension ou de suppression de la pension d'invalidité dans les conditions prévues aux articles L. 318 et L. 319 du Code de la sécurité sociale, la caisse primaire peut maintenir à l'invalide qui fait l'objet d'un traitement, suit des cours ou effectue un stage en vue de son reclassement ou de sa rééducation professionnelle, pendant toute la durée de ce traitement, cours ou stage, une fraction de ladite pension qui peut atteindre 50 p. 100.