أمثلة على PJRT

مثال: مكوّن CUDA الإضافي في JAX

1. تنفيذ واجهة برمجة التطبيقات PJRT C من خلال برنامج تضمين (pjrt_c_api_gpu.h).
2. إعداد نقطة دخول الحزمة (setup.py)
3. نفِّذ طريقة initialize()‎ (__init__.py).
4. يمكن اختبارها باستخدام أي اختبارات jax لوحدة معالجة الرسومات CUDA.

عمليات تنفيذ الأطر

في ما يلي بعض المراجع لاستخدام PJRT من جهة إطار العمل للتواصل مع أجهزة PJRT:

• JAX
  • يتفاعل jax-ml/jax مع واجهات برمجة تطبيقات PJRT من خلال واجهات برمجة تطبيقات xla_client.
• GoMLX
  • gomlx/gopjrt
  • gomlx/gomlx > backends/xla
• ZML
  • حزمة واجهة برمجة التطبيقات PJRT pjrt.zig
  • تحميل مكوّن PJRT الإضافي context.zig
  • التفاعل مع وحدات تخزين PJRT buffer.zig
  • تنفيذ وحدة عبر PJRT module.zig

عمليات تنفيذ الأجهزة

• الإضافات الكاملة للدمج (PJRT+MLIR+XLA):
  • مكوّن إضافي لوحدة المعالجة المركزية (XLA)
  • مكوّن إضافي لوحدة معالجة الرسومات في XLA
  • مكوّن Intel XLA الإضافي
• مكونات إضافية لدمج المحتوى البسيط (PJRT+MLIR):
  • المكوّن الإضافي StableHLO Reference Interpreter (مكوّن إضافي لـ C++ يستند إلى MLIR، وسيتم ربطه بعد devlabs)
  • مكوّن Tenstorrent-XLA الإضافي (مكوّن C الإضافي المستنِد إلى MLIR)