Code

Add aten::topk and its variants#547

Code

Merged

fengyuan14 merged 11 commits intomainintel/torch-xpu-ops:mainfrom xyt/topkintel/torch-xpu-ops:xyt/topkCopy head branch name to clipboardJul 23, 2024

+741-1

Commits

Commits on Jul 9, 2024

• add aten::topk

  xytintel

  committed
  796b2aa

  Copy full SHA for 796b2aa

Commits on Jul 11, 2024

• Merge branch 'main' into xyt/topk

  xytintel

  authored
  1933c2c

  Copy full SHA for 1933c2c

Commits on Jul 12, 2024

• Merge branch 'main' into xyt/topk

  xytintel

  authored
  bc1a3c7

  Copy full SHA for bc1a3c7

• Merge branch 'main' into xyt/topk

  xytintel

  authored
  89c712d

  Copy full SHA for 89c712d

Commits on Jul 15, 2024

• Merge branch 'main' into xyt/topk

  xytintel

  authored
  4f70965

  Copy full SHA for 4f70965

Commits on Jul 16, 2024

• Merge branch 'main' into xyt/topk

  xytintel

  authored
  3b5d06e

  Copy full SHA for 3b5d06e

• Using hard-coded work group size

  xytintel

  authored
  d263134

  Copy full SHA for d263134

• Merge branch 'main' into xyt/topk

  xytintel

  authored
  835d79a

  Copy full SHA for 835d79a

Commits on Jul 18, 2024

• handle k == 0

  xytintel

  authored
  d3beaaa

  Copy full SHA for d3beaaa

• Merge branch 'main' into xyt/topk

  xytintel

  authored
  b2a697e

  Copy full SHA for b2a697e

Commits on Jul 22, 2024

• Merge branch 'main' into xyt/topk

  xytintel

  authored
  31237bb

  Copy full SHA for 31237bb