Stage - Bac+5 - Evaluation d'Algorithmes en Précision Mixte sur Gpu pour un Solveur Dense H/F CEA

Des travaux précédents ont montré que l'introduction d'algorithmes en précision mixte dans le solveur permettent d'améliorer les performances tout en maintenant une précision suffisante. Cependant, les bibliothèques d'algèbre linéaire optimisées fournies par les vendeurs (e.g. cuBLAS/rocBLAS) n'offrent qu'un support très limité pour la précision mixte en arithmétique complexe.
Une première stratégie pour résoudre ce problème consiste en l'implémentation de noyaux en précision mixte complexe optimisés à l'aide de librairies telles que CUTLASS. La seconde stratégie envisagée consiste en la réécriture des algorithmes complexes en une succession de noyaux réels (approche 3M), qui nous permettrait donc de bénéficier des routines en précision mixte des bibliothèques des vendeurs. Le(la) candidat(e) aura pour mission d'implémenter ces stratégies dans le solveur afin d'évaluer leurs performances et leur précision.
En fonction du temps et des préférences du(de la) candidat(e), on pourra également explorer l'utilisation de méthodes d'émulation flottante de type BF16x9 ou Ozaki qui permettent d'émuler des hautes précisions à partir de calculs en faibles précisions.

Conformément aux engagements pris par le CEA en faveur de l'intégration des personnes handicapées, cet emploi est ouvert à toutes et à tous. Le CEA propose des aménagements et/ou des possibilités d'organisation pour l'inclusion des travailleurs handicapés.
Participant à la protection nationale, une enquête administrative est réalisée pour tous les collaborateurs du CEA afin d'assurer l'intégrité et la sécurité de la nation.