Enable native Kev inference with the AOTI CUDA backend
Chưa có ai nhận issue này.
Đánh giá
- Độ khó
- 5/5
- Thời gian dự kiến
- Hơn một tuần
- Mức phù hợp với người mới
- 30/100
- Loại issue
- Tính năng
- Độ rõ ràng
- Khá rõ ràng
- Mức độ hoạt động
- Sôi nổi
- Công nghệ
- cmake, python, pytorch
- Lĩnh vực
- ai-infra-agents, backend, machine-learning
Hướng nghiên cứu
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.
Do mô hình lập chỉ mục viết ra từ nội dung của issue.
Mô tả
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
prefillandscoremethods usingCudaPartitioner. Reuse the existing GDN prefill kernel, adapting layouts, gate/scaling conventions, and explicit initial/final state to Kev. - Wire up
aoti_cuda_backendin the example's CMake build and load the.pteplus CUDA delegate.ptddata throughModule. Handle device memory and synchronization when retaining prefix snapshots and reading scores on the host. - Preserve
system_one, explicitprefill/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_benchmarkto 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
- Ngôn ngữ chính
- Python
- Star
- 5k
- Fork
- 1.2k
- Merge trung bình
- 2 ngày 13 giờ
- Pull request đã merge (30 ngày)
- 595
Hướng dẫn đóng góp
Bắt đầu từ đâu
- Đọc hết issue, rồi đọc hướng dẫn đóng góp của dự án.
- Bình luận trên issue rằng bạn sẽ nhận — tránh hai người làm cùng một việc.
- Fork repository và làm thay đổi trên một nhánh.
- Mở pull request có tham chiếu số hiệu của issue.
Issue khác của pytorch/executorch
-
enhancement triaged
Độ khó 2/5 Nửa ngày Mức phù hợp với người mới 68/100
pytorch/executorch#21640 ·
-
good first issue module: examples module: webgpu
Độ khó 5/5 Hơn một tuần Mức phù hợp với người mới 35/100
pytorch/executorch#23099 ·
-
good first issue module: examples module: vulkan
Độ khó 5/5 Hơn một tuần Mức phù hợp với người mới 30/100
pytorch/executorch#23098 ·
-
module: qnn partner: qualcomm
pytorch/executorch#23096 · 1 bình luận · 1 reaction · 1 người được giao ·
-
pytorch/executorch#23058 · 1 reaction · 1 người được giao ·
Tất cả issue của pytorch/executorch
Issue tương tự
-
[Bug] reef-hermes tells me to resume with hermes --resume, which does not work from my shell Đang mởarea: harness bug status: needs-triage
Độ khó 2/5 1-3 giờ Mức phù hợp với người mới 75/100
Human-Agent-Society/reef#625 ·
-
Độ khó 2/5 1-3 giờ Mức phù hợp với người mới 70/100
-
Độ khó 1/5 Dưới một giờ Mức phù hợp với người mới 80/100
learningequality/kolibri#15351 · 2 bình luận ·
-
Độ khó 2/5 1-3 giờ Mức phù hợp với người mới 75/100
-
Name consistency Đang mở
Độ khó 2/5 1-3 giờ Mức phù hợp với người mới 75/100
eellak/triplestore#65 · 1 bình luận ·