Code

Add topk operator#66

Code

Merged

josel-amd merged 11 commits intofeature/fused-opsXilinx/mlir-xten:feature/fused-opsfrom matthias.topkXilinx/mlir-xten:matthias.topkCopy head branch name to clipboardAug 21, 2024

+192-18

Commits

Commits on Aug 15, 2024

• Add topk operator

  mgehre-amd

  committed
  a2b0fd6

  Copy full SHA for a2b0fd6

Commits on Aug 20, 2024

• Add the Indice to the outputs

  Show description for c9dfb61
  josel-amd

  committed
  c9dfb61

  Copy full SHA for c9dfb61

• Use K as an operand instead of attribute

  Show description for 37321a5
  josel-amd

  committed
  37321a5

  Copy full SHA for 37321a5

• Do not make constant k mandatory

  josel-amd

  committed
  dd3f01a

  Copy full SHA for dd3f01a

• Test topk with an argument

  josel-amd

  committed
  5b11a93

  Copy full SHA for 5b11a93

• Add support for dynamic topk

  josel-amd

  committed
  2316e7e

  Copy full SHA for 2316e7e

• k must be <= dimSize

  josel-amd

  committed
  e3b57e5

  Copy full SHA for e3b57e5

• Remove unused include

  josel-amd

  committed
  9d060f9

  Copy full SHA for 9d060f9

Commits on Aug 21, 2024

• Relax result type comparison

  Show description for 9939251
  josel-amd

  committed
  9939251

  Copy full SHA for 9939251

• Add a test case

  josel-amd

  committed
  81503ea

  Copy full SHA for 81503ea

• Fix failing test

  josel-amd

  committed
  e995188

  Copy full SHA for e995188