Code

Register grad for torch.Tensor.item#1481

Code

Merged

IvanYashchuk merged 3 commits intomainLightning-AI/lightning-thunder:mainfrom

item-vjpLightning-AI/lightning-thunder:item-vjpCopy head branch name to clipboard

Nov 28, 2024

Commits

Commits on Nov 27, 2024

• Add grad rule for torch.Tensor.item()

  IvanYashchuk

  committed
  073d34e

  Copy full SHA for 073d34e

• Add a test

  IvanYashchuk

  committed
  3ac12fb

  Copy full SHA for 3ac12fb

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

  Show description for 0c07ef7
  pre-commit-ci[bot]

  committed
  0c07ef7

  Copy full SHA for 0c07ef7