يستكشف برنامج تعليمي جديد إطار عمل TileGym الخاص بشركة NVIDIA من خلال بناء سير عمل عملي لـ Colab يتكيف مع ظروف الأجهزة المختلفة. يفحص البرنامج التعليمي بيئة CUDA لتحديد ما إذا كان يمكن تشغيل NVIDIA cuTile مباشرة، مع العودة إلى Triton عندما تفتقر وحدات معالجة الرسومات القياسية في Colون إلى المكدس المطلوب.
- يعمل نموذج برمجة البلاط الأساسي على بلاطات بيانات كاملة بدلاً من الخيوط الفردية، حيث يتم تحميلها داخل النواة لحساب فعال.
- يقوم البرنامج التعليمي بتنفيذ جمع المتجهات وGELU المدمج وsoftmax لكل صف وضرب المصفوفات بالبلاط وflash attention باستخدام بداهيات cuTile وTriton معًا.
- تتم مقارنة كل تنفيذ مع PyTorch من حيث الدقة ويتم قياس الأداء.
تعمل هذه الموارد كدليل عملي لفهم مفاهيم البرمجة القائمة على البلاط عبر قدرات وحدات معالجة الرسومات المختلفة.