講演情報
[B-6-83]Python科学計算向けCPU/GPUバックエンド選択・コード変換手法の提案
〇黄 掣1、岩城 亜弥子1、秦野 智也1、島田 達也1 (1. NTT株式会社)
キーワード：
処理時間予測、CPU/GPUバックエンド選択、コード変換
クラウド環境でのPython科学計算では、CPU向けNumPyとGPU向けCuPyの使い分けが性能に大きく影響するが、環境や入力サイズに応じた手動調整が必要である。本稿では、NumPy/CuPy共通関数を対象に、関数単位の実行時間予測とCPU–GPU間転送時間を考慮したバックエンド選択、およびAST解析によるソースコード自動変換手法を提案する。3種類のアクセラレータ環境で評価した結果、平均MAPEは1.57〜1.62%であり、約99%の予測対象でMAPE 10%以下を達成した。また、静的書換えおよび実行時CPU/GPU自動切替により、固定的なバックエンド選択より平均処理時間を短縮できることを確認した。