Code

Max pool#163

Code

Merged

t-vi merged 23 commits intoLightning-AI:mainLightning-AI/lightning-thunder:mainfrom jjsjann123:max_pooljjsjann123/lightning-thunder:max_poolCopy head branch name to clipboardApr 12, 2024

+152-2

Commits

Commits on Apr 11, 2024

• special grad_transform for max_poolxd

  jjsjann123

  committed
  1756c03

  Copy full SHA for 1756c03

• forgot the name

  jjsjann123

  committed
  9511603

  Copy full SHA for 9511603

• adding torch symbol for max_pool backward

  jjsjann123

  committed
  f641130

  Copy full SHA for f641130

• fixing signature

  jjsjann123

  committed
  75e284c

  Copy full SHA for 75e284c

• removing 3d because of pytorch aten API coverage

  jjsjann123

  committed
  cc489c6

  Copy full SHA for cc489c6

• fixing max_pool2d with indices

  jjsjann123

  committed
  486060a

  Copy full SHA for 486060a

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for d252018
  pre-commit-ci[bot]

  committed
  d252018

  Copy full SHA for d252018

• adding torch operator max_pool2d_with_indices

  jjsjann123

  committed
  d614035

  Copy full SHA for d614035

• Merge remote-tracking branch 'jiej/max_pool' into max_pool

  jjsjann123

  committed
  104a96e

  Copy full SHA for 104a96e

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for c995511
  pre-commit-ci[bot]

  committed
  c995511

  Copy full SHA for c995511

Commits on Apr 12, 2024

• patch backward operator

  jjsjann123

  committed
  82f56e9

  Copy full SHA for 82f56e9

• patch

  jjsjann123

  committed
  22efa40

  Copy full SHA for 22efa40

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for 254fcb0
  pre-commit-ci[bot]

  committed
  254fcb0

  Copy full SHA for 254fcb0

• fixing logic

  jjsjann123

  committed
  0e9c441

  Copy full SHA for 0e9c441

• functionally correct now at least

  jjsjann123

  committed
  d8647bb

  Copy full SHA for d8647bb

• Merge branch 'main' into max_pool

  jjsjann123

  authored
  e9e9dd9

  Copy full SHA for e9e9dd9

• refactor to support max_pool3d as well

  jjsjann123

  committed
  7a06c07

  Copy full SHA for 7a06c07

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for ef7ae3a
  pre-commit-ci[bot]

  committed
  ef7ae3a

  Copy full SHA for ef7ae3a

• partial can't be used in grad_transform

  jjsjann123

  committed
  351469d

  Copy full SHA for 351469d

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for f8a167f
  pre-commit-ci[bot]

  committed
  f8a167f

  Copy full SHA for f8a167f

• addressing reviews

  jjsjann123

  committed
  dfaa058

  Copy full SHA for dfaa058

• typo

  jjsjann123

  committed
  155ac78

  Copy full SHA for 155ac78

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for 2a8da1b
  pre-commit-ci[bot]

  committed
  2a8da1b

  Copy full SHA for 2a8da1b