Code

Add aten::index_reduce operator#1156

Code

Open

cfgfung wants to merge 6 commits intomainintel/torch-xpu-ops:mainfrom reduce_index_v2intel/torch-xpu-ops:reduce_index_v2Copy head branch name to clipboard

+507-1

Commits

Commits on Nov 25, 2024

• First reduced version of index_reduce.

  cfgfung

  committed
  92738af

Commits on Nov 26, 2024

• Implemented reduce_prod.

  cfgfung

  committed
  cbd7b03

Commits on Dec 2, 2024

• Removed unnecessary if cases.

  cfgfung

  committed
  55daf53

Commits on Dec 3, 2024

• Added two reduce operators - amin and amax.

  cfgfung

  committed
  eb1ad53

Commits on Dec 4, 2024

• Add reduce_mean op.

  cfgfung

  committed
  f77e325

Commits on Dec 19, 2024

• Skip 3 test cases. These are due to precision errors and the difference is very small.

  cfgfung

  committed
  9066b5d