Code

Add torchcompile_xentropy executor#1655

Code

Draft

riccardofelluga wants to merge 8 commits intomainLightning-AI/lightning-thunder:mainfrom torchcompilecat-add-xentropyLightning-AI/lightning-thunder:torchcompilecat-add-xentropyCopy head branch name to clipboard

+58-2

Commits

Commits on Jan 17, 2025

• add cross_entropy to torchcompile_cat executor

  riccardofelluga

  committed
  6c5b0e1

  Copy full SHA for 6c5b0e1

Commits on Jan 28, 2025

• Merge branch 'main' of github.com:Lightning-AI/lightning-thunder into torchcompilecat-add-xentropy

  riccardofelluga

  committed
  3982cb7

  Copy full SHA for 3982cb7

• Revert "add cross_entropy to torchcompile_cat executor"

  Show description for ecc1f7c
  riccardofelluga

  committed
  ecc1f7c

  Copy full SHA for ecc1f7c

• move xentropy in its own executor

  riccardofelluga

  committed
  0927d74

  Copy full SHA for 0927d74

• reshape not always required

  riccardofelluga

  committed
  e11ddab

  Copy full SHA for e11ddab

• add torchcompile_xentropy to the tests

  riccardofelluga

  committed
  d815892

  Copy full SHA for d815892

• add torchcompile_xentropy as default executor

  riccardofelluga

  committed
  39530d3

  Copy full SHA for 39530d3

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

  Show description for 284e7d9
  pre-commit-ci[bot]

  committed
  284e7d9

  Copy full SHA for 284e7d9