Code

Add bitonic topk#3862

Code

Open

pfultz2 wants to merge 39 commits intodevelopROCm/AMDMIGraphX:developfrom bitonic-topk2ROCm/AMDMIGraphX:bitonic-topk2Copy head branch name to clipboard

+1,297-112

Commits

Commits on Jan 30, 2025

• Add bitonic topk

  pfultz2

  committed
  878f970

  Copy full SHA for 878f970

Commits on Jan 31, 2025

• Refactor

  pfultz2

  committed
  a992475

  Copy full SHA for a992475

• Adjust tile size

  pfultz2

  committed
  da2d455

  Copy full SHA for da2d455

Commits on Feb 3, 2025

• Add wave sort

  pfultz2

  committed
  be70a65

  Copy full SHA for be70a65

• Compare backwards

  pfultz2

  committed
  f39178a

  Copy full SHA for f39178a

Commits on Feb 4, 2025

• Try to use block topk

  pfultz2

  committed
  41d06c3

  Copy full SHA for 41d06c3

• Some mroe bug fixes

  pfultz2

  committed
  795f6f5

  Copy full SHA for 795f6f5

• Add compare HOF

  pfultz2

  committed
  4c3a387

  Copy full SHA for 4c3a387

Commits on Feb 5, 2025

• Add split topk

  pfultz2

  committed
  6bd26a0

  Copy full SHA for 6bd26a0

Commits on Feb 6, 2025

• Adjust min_size

  pfultz2

  committed
  4742fe9

  Copy full SHA for 4742fe9

Commits on Feb 7, 2025

• Adjust the split

  pfultz2

  committed
  7a5c65c

  Copy full SHA for 7a5c65c

• Return bool

  pfultz2

  committed
  7a3c569

  Copy full SHA for 7a3c569

Commits on Feb 9, 2025

• Fix bug

  pfultz2

  committed
  1c621c0

  Copy full SHA for 1c621c0

• Fix out of bounds access

  pfultz2

  committed
  a055fd6

  Copy full SHA for a055fd6

Commits on Feb 10, 2025

• Handle larger sizes correctly

  pfultz2

  committed
  994206f

  Copy full SHA for 994206f

• Add tests

  pfultz2

  committed
  af4b374

  Copy full SHA for af4b374

• Format

  pfultz2

  committed
  9909e75

  Copy full SHA for 9909e75

• Update license

  pfultz2

  committed
  738e30c

  Copy full SHA for 738e30c

Commits on Feb 11, 2025

• Fix perf for smaller size

  pfultz2

  committed
  9d8d203

  Copy full SHA for 9d8d203

• Remove standard requirement and increase batch in test

  pfultz2

  committed
  2796b25

  Copy full SHA for 2796b25

• Set index type

  pfultz2

  committed
  84a8470

  Copy full SHA for 84a8470

• Pack pair

  pfultz2

  committed
  f6cdd43

  Copy full SHA for f6cdd43

Commits on Feb 27, 2025

• Improve constants

  pfultz2

  committed
  ebd3657

  Copy full SHA for ebd3657

• Use slice header

  pfultz2

  committed
  cc1aef3

  Copy full SHA for cc1aef3

• Fix ctad

  pfultz2

  committed
  87116da

  Copy full SHA for 87116da

• Add sort header

  pfultz2

  committed
  e4b9ea5

  Copy full SHA for e4b9ea5

• Use unrolled repeat

  pfultz2

  committed
  09ecb57

  Copy full SHA for 09ecb57

• Move var

  pfultz2

  committed
  547d1d6

  Copy full SHA for 547d1d6

• Use repeat_by_2 functions instead of recursion

  pfultz2

  committed
  bf58ba9

  Copy full SHA for bf58ba9

• Move to funcitonal

  pfultz2

  committed
  0e7f659

  Copy full SHA for 0e7f659

Commits on Feb 28, 2025

• Only wave sort when it reduces final topk sort

  pfultz2

  committed
  dad9fe3

  Copy full SHA for dad9fe3

• Use strided loads

  pfultz2

  committed
  a4324a0

  Copy full SHA for a4324a0

Commits on Mar 1, 2025

• Update ref version of topk to handle input indices

  pfultz2

  committed
  3364997

  Copy full SHA for 3364997

• Update ref tests

  pfultz2

  committed
  6600a55

  Copy full SHA for 6600a55

• Add test for axis=0

  pfultz2

  committed
  9e9329c

  Copy full SHA for 9e9329c

Commits on Mar 3, 2025

• Use slicing

  pfultz2

  committed
  f14a957

  Copy full SHA for f14a957

• Format

  pfultz2

  committed
  f04b592

  Copy full SHA for f04b592

• Update license

  pfultz2

  committed
  afd9f00

  Copy full SHA for afd9f00

• Format

  pfultz2

  committed
  c5a34d0

  Copy full SHA for c5a34d0