Repository navigation

Exposes `DeviceBatchedTopK::{Min,Max}{Keys,Pairs}` for non-deterministic, unordered, and small segments-only - #9331

Merged

elstehle merged 23 commits into

NVIDIA:mainfrom

elstehle:topk/expose-device-segmented-topk

Jul 1, 2026

Merged

Exposes `DeviceBatchedTopK::{Min,Max}{Keys,Pairs}` for non-deterministic, unordered, and small segments-only#9331
elstehle merged 23 commits into
NVIDIA:mainfrom
elstehle:topk/expose-device-segmented-topk

Commits on Jun 22, 2026
- exposes batched top-k for small segs
  elstehle
  committed
  6eeb08f
- Document DeviceBatchedTopK determinism, tie-break, and output ordering
  elstehle
  committed
  8c8e07a
- extends top-k requirements page for DeviceBatchedTopK
  elstehle
  committed
  536e0e2
- improves docs and examples
  elstehle
  committed
  bad1e4c
- better cuda::args docs
  elstehle
  committed
  ca103b6
- drops standalone examples for follow-up pr
  elstehle
  committed
  fbd66d7
- addresses review comments
  elstehle
  committed
  b6c8113
- refines docs
  elstehle
  committed
  8111d1b
Commits on Jun 23, 2026
- introduces the both-or-nothing rule for determinism and tie-break
  elstehle
  committed
  2f7f695
- adds documentation on function parameters
  elstehle
  committed
  d851e22
- validates unwrapped parameters
  elstehle
  committed
  c777859
- clarifies support for plain integrals
  elstehle
  committed
  6cb0864
- fixes copydetails
  elstehle
  committed
  b9ffd43
- Merge remote-tracking branch 'upstream/main' into topk/expose-device-segmented-topk
  elstehle
  committed
  ef9d2b6
Commits on Jun 24, 2026
- Merge remote-tracking branch 'upstream/main' into topk/expose-device-segmented-topk
  elstehle
  committed
  3625924
Commits on Jun 27, 2026
- Merge remote-tracking branch 'upstream/main' into topk/expose-device-segmented-topk
  elstehle
  committed
  c1226fb
- adds api test for unwrapped integrals
  elstehle
  committed
  4cbb929
- addresses review comments
  elstehle
  committed
  6b0e679
Commits on Jun 30, 2026

Navigation Menu

Repository navigation

Exposes `DeviceBatchedTopK::{Min,Max}{Keys,Pairs}` for non-deterministic, unordered, and small segments-only - #9331

Exposes `DeviceBatchedTopK::{Min,Max}{Keys,Pairs}` for non-deterministic, unordered, and small segments-only#9331
elstehle merged 23 commits into
NVIDIA:mainfrom
elstehle:topk/expose-device-segmented-topk

Commits on Jun 22, 2026

exposes batched top-k for small segs

Document DeviceBatchedTopK determinism, tie-break, and output ordering

extends top-k requirements page for DeviceBatchedTopK

improves docs and examples

better cuda::args docs

drops standalone examples for follow-up pr

addresses review comments

refines docs

Commits on Jun 23, 2026

introduces the both-or-nothing rule for determinism and tie-break

adds documentation on function parameters

validates unwrapped parameters

clarifies support for plain integrals

fixes copydetails

Merge remote-tracking branch 'upstream/main' into topk/expose-device-segmented-topk

Commits on Jun 24, 2026

Merge remote-tracking branch 'upstream/main' into topk/expose-device-segmented-topk

Commits on Jun 27, 2026

Merge remote-tracking branch 'upstream/main' into topk/expose-device-segmented-topk

adds api test for unwrapped integrals

addresses review comments

Commits on Jun 30, 2026

Merge remote-tracking branch 'upstream/main' into topk/expose-device-segmented-topk

adds to the umbrella cub.cuh and device-wide docs

const-rev env and docs on in/out overlapping

improves api examples; adds mis-aligned temp storage test

Merge remote-tracking branch 'upstream/main' into topk/expose-device-segmented-topk