Code

Add torchcompile_xentropy executor#1655

Code

Merged

t-vi merged 22 commits intomainLightning-AI/lightning-thunder:mainfrom torchcompilecat-add-xentropyLightning-AI/lightning-thunder:torchcompilecat-add-xentropyCopy head branch name to clipboardFeb 9, 2025

+81-3

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

Commits on Jan 29, 2025

• remove torch.nn.functional.cross_entropy from supported as it is in required

  riccardofelluga

  committed
  c4d98e9

  Copy full SHA for c4d98e9

• bump version (#1708)

  

  ali-alshaar7

  authored and

  riccardofelluga

  committed
  9338118

  Copy full SHA for 9338118

• nvfuser: add option to allow shape only region (#1702)

  

  kshitij12345

  authored and

  riccardofelluga

  committed
  c875621

  Copy full SHA for c875621

• Backward transform dependency fix (#1693)

  

  jjsjann123

  authored and

  riccardofelluga

  committed
  7e857f5

  Copy full SHA for 7e857f5

• pin check schema (#1709)

  

  t-vi

  authored and

  riccardofelluga

  committed
  fcec023

  Copy full SHA for fcec023

• bump transformers version (#1698)

  riccardofelluga

  committed
  9a40bb0

  Copy full SHA for 9a40bb0

• Reduce test_thunderfx_mistral_nemo_small model size (#1701)

  Show description for 1df5929
  riccardofelluga

  committed
  1df5929

  Copy full SHA for 1df5929

• fix naming

  riccardofelluga

  committed
  71bd489

  Copy full SHA for 71bd489

• add pad prims

  riccardofelluga

  committed
  d94a420

  Copy full SHA for d94a420

• add test for the new executor

  riccardofelluga

  committed
  35c4606

  Copy full SHA for 35c4606

• Merge branch 'main' into torchcompilecat-add-xentropy

  riccardofelluga

  authored
  e0e32e0

  Copy full SHA for e0e32e0

• add missing import

  riccardofelluga

  committed
  3edc0a3

  Copy full SHA for 3edc0a3

• Merge branch 'main' into torchcompilecat-add-xentropy

  riccardofelluga

  authored
  3cebfe3

  Copy full SHA for 3cebfe3

Commits on Feb 4, 2025

• Merge branch 'main' into torchcompilecat-add-xentropy

  riccardofelluga

  authored
  52294ec

  Copy full SHA for 52294ec