arXiv cs.AI

KForge: 多様なAIハードウェアアクセラレーター向けプログラム合成

KForge: Program Synthesis for Diverse AI Hardware Accelerators

http://arxiv.org/abs/2511.13274v1


KForgeは、異なるAIハードウェアアクセラレーター向けに最適化されたプログラム合成を行うプラットフォーム非依存のフレームワークです。このフレームワークは、生成エージェントとパフォーマンス分析エージェントの2つのLLM(大規模言語モデル)ベースのエージェントに基づいて構築されています。生成エージェントはコンパイルと正確性のフィードバックを通じてプログラムを生成し、繰り返し改良し、パフォーマンス分析エージェントはプロファイリングデータを解釈して最適化を導く役割を果たします。KForgeは、新しいプラットフォームを標的とするために単一の例で動作することができます。本研究は、異なるアクセラレーター向けにプログラム合成をガイドするための実用的な推薦を生成するイテレーティブな改良システムを紹介し、また、異なるハードウェアターゲットのための生成品質を向上させるクロスプラットフォームの知識移転を効果的に活用することを示しています。さらに、CUDAとApple Metalという異なる並列計算プラットフォームでのプログラム合成の有効性を検証しています。