Stage Optimisation FFTs
REF : FFT
Montbonnot / STAGE
Kalray recherche
Un(e) stagiaire pour optimiser des FFTs en exploitant les opérateurs matriciels FP16/FP32 des processeurs MPPA
REF : FFT
SUJET
Rattaché(e) à l’équipe du CTO vous participez à l’amélioration des performances de transformées de Fourier rapides (FFT) sur les cœurs des processeurs de Kalray. Il s’agit d’une opportunité de découvrir ou d’enrichir vos connaissances en algorithmie et implémentation en C des transformées de Fourier rapides.
Kalray produit le processeur MPPA3 qui comporte 80 cœurs applicatifs d’architecture 64-bit VLIW appelée KVX répartis en cinq ‘clusters’ de 16 cœurs SMP. Chaque cœur KVX est programmable en C/C++/OpenMP dans un environnement de compilateurs GCC et LLVM, de GDB et autres outils GNU. Chaque cœur intègre un opérateur de multiplication addition de matrices 4×8 par 8×4 d’éléments FP16 vers une matrice 4×4 FP32, dont les instructions sont accessibles depuis GCC et LLVM au travers de types et d’opérations intrinsèques.
Les objectifs du stage consistent à implementer des FFTs 1D et 2D opérant sur données FP16 et s’appuyant sur les opérateurs matriciels des cœurs KVX. Des transformées de Fourier rapides en FP32 ont déjà été développées par Kalray pour ces cœurs : “In-Place Multicore SIMD Fast Fourier Transforms” https://inria.hal.science/hal-04240798 mais sans faire intervenir les opérateurs de multiplication addition de matrices. En revanche, les pistes algorithmiques à suivre pour ce type d’implémentation ont été établies dans un rapport technique.
A ce titre, vos missions principales sont les suivantes :
- Prendre en main l’environnement GCC avec types et opérations matricielles pour les coeurs KVX.
- Produire des implémentations fonctionnelles de FFT 1D puis 2D en C sur données FP16.
- Optimiser ces implémentations en exploitant les opérateurs de multiplication addition de matrices.
COMPÉTENCES REQUISES ET À DÉVELOPPER
- Compréhension des techniques d’implémentation, d’optimisation SIMD et de parallélisation des FFTs.
- Alignement des interfaces sur les standard FFT de l’industrie (ex. CMSIS-DSP de ARM) et productisation.
CONTRAT
- Type de contrat : convention de stage M1, M2 ou PFE.
- Gratification selon grille
- Convention collective : Métallurgie.
- Avantages : Titres restaurant, CE, événements…
- Lieu : Montbonnot (38).
A PROPOS DE KALRAY
Kalray est une des plus prometteuses et innovantes sociétés de la « DeepTech » française, essaimage du CEA. Leader dans les technologies matérielles et logicielles dédiées à la gestion et au traitement intensif des données du Cloud au Edge, Kalray propose une gamme complète de produits permettant de développer des applications, ainsi que des infrastructures, plus intelligentes, plus efficaces et plus économes en énergie. L’offre de Kalray comprend des processeurs pluri-coeurs, des cartes d’accélération, ainsi que des solutions logicielles de stockage et de gestion des données pour des marchés en fort développement : datacenters, IA, industrie des médias et du divertissement, sciences de la vie, recherche scientifique, Edge computing, automobile…
En forte croissance, Kalray compte près de 200 collaborateurs/trices. La société est présente en France, au Royaume-Uni, en Allemagne, aux Etats-Unis ainsi qu’au Japon et accueille à son capital des investisseurs de premier plan comme l’Alliance Venture (Renault-Nissan-Mitsubishi), NXP Semiconductors, le CEA et Bpifrance. Kalray est également cotée en bourse sur Euronext Growth (FR0010722819 – ALKAL).
Kalray, c’est avant tout une équipe de femmes et d’hommes passionnés, enthousiastes et solidaires, qui mettent l’excellence et la collaboration au cœur de leur quotidien. Que vous soyez expert de votre domaine ou jeune diplômé, vous saurez trouver votre place et vous épanouir dans un environnement agile, dynamique et handi-bienveillant, où vous pourrez pleinement exprimer votre talent et contribuer très concrètement au succès de l’entreprise. Au programme : des challenges passionnants, mais aussi de la bonne humeur, de l’entraide, la montagne, la mer… et bien plus encore !
