Skip to content

b11632

Pre-release
Pre-release

Choose a tag to compare

@github-actions github-actions released this 15 Oct 03:20
62df181

cuda, hip: GDN chunked kernel (#29353)

  • cuda, hip: GDN chunked kernel

Ported-from: tetherto#257
Source-commit: 53f0027
Co-authored-by: Aman Gupta aman.gupta@tether.io

  • fix older builds

  • review fixes

  • cuda: enable GDN MMA on tested Ampere and Ada GPUs

Enable SM86 and SM89 after RTX 3090 and RTX 4090 correctness and prefill benchmarks. Keep other untested architectures on the existing fallback.

Assisted-by: Codex

  • cuda : use refactored swizzle API in GDN MMA

  • ggml : GDN only guarantees the final state when n_tokens > K

  • cuda : add short GDN MMA phase comments


Co-authored-by: gianni-cor gianfranco.cordella@tether.io
Co-authored-by: Aman Gupta aman.gupta@tether.io

Website:

Attestations:

macOS/iOS:

Linux:

Android:

Windows:

openEuler:

  • DISABLED
  • openEuler x86 (310p)
  • openEuler x86 (910b, ACL Graph)
  • openEuler aarch64 (310p)
  • openEuler aarch64 (910b, ACL Graph)

UI: