Hacktoberfest 2026: những issue maintainer đã đánh dấu cho tháng Mười, đang mở và phù hợp người mới. Xem issue Hacktoberfest

[Feature Request] Add Gated Delta Net (GDN) support

Đã đóng
  • Có gì không ổn?
#2,884 0 bình luận 0 reaction 0 người được giao Xem trên GitHub

Maintainer thường phản hồi trong vòng 2 ngày

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
25/100
Loại issue
Tính năng
Độ rõ ràng
Cần làm rõ
Mức độ hoạt động
Ít trao đổi
Công nghệ
python
Lĩnh vực
machine-learning, performance

Hướng nghiên cứu

Bắt đầu bằng cách xác định layer GDN hiện có cùng các path fla.ops.gated_delta_rule và causal_conv1d của nó. Xác định phạm vi của native TE kernel cho chunked training, inference, packed sequences, BF16 và FP8, sau đó benchmark hiệu năng training trên GB200. Được xem là hoàn tất khi Megatron Core sử dụng native path mà không có các dependency bên ngoài đó và bao phủ chức năng được yêu cầu.

Do mô hình lập chỉ mục viết ra từ nội dung của issue.

Mô tả

Is your feature request related to a problem? Please describe.

Megatron Core already has a Gated Delta Net (GDN) layer, but the current implementation depends on external Triton-based fla gated-delta-rule kernels and causal_conv1d. This means GDN does not benefit from native TE kernel performance, and users need extra dependencies.

The current training kernel path is also not performing well in practice, and the gap is noticeable on GB200.

In addition, the current path still has functional gaps: inference is not supported and packed sequences are not supported.

Describe the solution you'd like

Add native TE kernels for Gated Delta Net (GDN), with Megatron Core integration.

Ideally this would include:

  • an optimized kernel path for the chunked gated delta rule
  • substantially better training performance, especially on GB200
  • compatibility with TE mixed-precision flows, especially BF16 and FP8

Describe alternatives you've considered

Using the current Triton-based fla.ops.gated_delta_rule + causal_conv1d implementation in Megatron Core.

Ngôn ngữ chính
Python
Star
3.6k
Fork
862
Merge trung bình
4 ngày 18 giờ
Pull request đã merge (30 ngày)
64

Chuẩn bị môi trường

Bắt đầu từ đâu

  1. Đọc hết issue, rồi đọc hướng dẫn đóng góp của dự án.
  2. 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.
  3. Fork repository và làm thay đổi trên một nhánh.
  4. Mở pull request có tham chiếu số hiệu của issue.

Issue khác của NVIDIA/TransformerEngine

Tất cả issue của NVIDIA/TransformerEngine

Issue tương tự

Thêm issue về Python

Nhận issue mới trong hộp thư của bạn

Bản tóm tắt ngắn những issue GitHub phù hợp với người mới.