Hacktoberfest 2026:メンテナが10月に向けて印を付けた、オープンで初心者向けの issue。 Hacktoberfest の issue を見る

Enable native Kev inference with the AOTI CUDA backend

オープン
#23,097 コメント 0 件 リアクション 0 件 担当者 0 名 GitHub で見る

まだ誰も着手していません。

評価

難易度
5/5
見積もり時間
1週間以上
初心者へのやさしさ
30/100
issue の種類
機能追加
明瞭さ
おおむね明確
活発さ
活発
技術スタック
cmake, python, pytorch

調査の方向性

Start by examining the existing native Kev example and the Qwen3.5 MoE CUDA export reference. Focus on adapting the CudaPartitioner for Kev's prefill and score methods. Study the chunked GDN kernel and backend tests. The goal is to integrate the aoti_cuda_backend into the example's CMake build, handle device memory, and validate against the PyTorch implementation. Ensure the runner and benchmark are reused.

索引モデルが issue の本文から書いたものです。

説明

good first issue module: cuda module: examples

Enable the native Kev example on NVIDIA GPUs through ExecuTorch's AOTI CUDA backend. The example landed in #23023 with XNNPACK and MLX support.

Kev runs prefill followed by a pointer head over option-boundary hidden states. A prefix snapshot is reused across question batches. The goal is to run this same workflow through the existing C++ API, without Python at inference.

Use Qwen3.5 MoE as the primary reference for CUDA export, native loading, and Gated DeltaNet (GDN). Its chunked GDN kernel and backend tests are useful starting points.

Work to cover:

  • Add a CUDA export option for Kev's prefill and score methods using CudaPartitioner. Reuse the existing GDN prefill kernel, adapting layouts, gate/scaling conventions, and explicit initial/final state to Kev.
  • Wire up aoti_cuda_backend in the example's CMake build and load the .pte plus CUDA delegate .ptd data through Module. Handle device memory and synchronization when retaining prefix snapshots and reading scores on the host.
  • Preserve system_one, explicit prefill/evaluate, configurable token limits, variable question/option counts, and batching beyond eight questions. Repeated evaluations must leave the prefix unchanged.
  • Support unquantized BF16 weights while preserving FP32 recurrence, pointer-head scoring, fitted temperature, and checkpoint metadata. Compare logits and probabilities against upstream Kev's PyTorch implementation and the existing FP32 path, with documented tolerances.
  • Validate varying batch/sequence shapes, including GDN chunk boundaries and repeated prefix reuse. Add reproducible export/build/run commands and measurements using kev_benchmark to the README; verify the backbone actually runs on CUDA and report any CPU fallback.

Keep the integration in the existing example and reuse the runner and benchmark. An NVIDIA GPU and CUDA development environment are needed for validation.

cc @iseeyuan @lucylq @helunwencser @tarun292 @kimishpatel @jackzhxng @Gasoonjia @digantdesai

主要言語
Python
スター
5k
フォーク
1.2k
平均マージ
2日 13時間
マージ済み PR(30日)
595

コントリビューションガイド

コントリビューションガイドを開く

はじめの一歩

  1. issue を最後まで読み、次にプロジェクトのコントリビューションガイドを読みます。
  2. 着手することを issue にコメントします — 二人が同じ作業をするのを防げます。
  3. リポジトリをフォークし、ブランチを切って変更します。
  4. issue 番号を参照したプルリクエストを送ります。

pytorch/executorch のほかの issue

pytorch/executorch の issue をすべて見る

似ている issue

Python の issue をもっと見る

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。