Code

Adding shape prim#1113

Code

Merged

t-vi merged 12 commits intomainLightning-AI/lightning-thunder:mainfrom shape_primLightning-AI/lightning-thunder:shape_primCopy head branch name to clipboardSep 11, 2024

+52-5

Commits

Commits on Sep 6, 2024

• initial commit

  jjsjann123

  committed
  52fc1fa

  Copy full SHA for 52fc1fa

• fixing small things

  jjsjann123

  committed
  9001522

  Copy full SHA for 9001522

• fixing nvfuser impl for size

  jjsjann123

  committed
  6317efe

  Copy full SHA for 6317efe

• adding no-op grad rule

  jjsjann123

  committed
  4a39f41

  Copy full SHA for 4a39f41

• fixing executor handle of size op

  jjsjann123

  committed
  2eebbc6

  Copy full SHA for 2eebbc6

• Merge remote-tracking branch 'origin/main' into HEAD

  jjsjann123

  committed
  e8841a2

  Copy full SHA for e8841a2

• renaming size->shape in prim

  jjsjann123

  committed
  f1325af

  Copy full SHA for f1325af

• size->shape

  jjsjann123

  committed
  e9de3ef

  Copy full SHA for e9de3ef

Commits on Sep 8, 2024

• Merge branch 'main' into shape_prim

  jjsjann123

  authored
  b3f67a5

  Copy full SHA for b3f67a5

• changing len(tensor.size()) to tensor.ndim

  jjsjann123

  committed
  a61f88d

  Copy full SHA for a61f88d

Commits on Sep 9, 2024

• thanks to TomV

  jjsjann123

  committed
  545d384

  Copy full SHA for 545d384

• removing torchsymbol for size

  jjsjann123

  committed
  8c67ef6

  Copy full SHA for 8c67ef6