Code

Add PyTorch 1.8 Profiler 5/5#6618

Code

Merged

tchaton merged 153 commits intomasterLightning-AI/pytorch-lightning:masterfrom update-pytorch-profiler-1.8Lightning-AI/pytorch-lightning:update-pytorch-profiler-1.8Copy head branch name to clipboardMar 23, 2021

+399-96

Commits

Commits on Mar 4, 2021

• Refactor profilers

  carmocca

  committed
  aa89fa1

  Copy full SHA for aa89fa1

Commits on Mar 5, 2021

• Update PassThrough

  carmocca

  committed
  b66f1f6

  Copy full SHA for b66f1f6

Commits on Mar 9, 2021

• WIP - This is broken and will change

  carmocca

  committed
  d388d80

  Copy full SHA for d388d80

• Update pytorch_lightning/profiler/pytorch.py

  Show description for 93aa073
  

  carmocca

  and

  tchaton

  authored
  93aa073

  Copy full SHA for 93aa073

Commits on Mar 18, 2021

• resolve tests

  tchaton

  committed
  0f8b6ef

  Copy full SHA for 0f8b6ef

• Merge branch 'master' into update-pytorch-profiler

  tchaton

  committed
  d219ae1

  Copy full SHA for d219ae1

• resolve tests

  tchaton

  committed
  8a3d3cf

  Copy full SHA for 8a3d3cf

• find output

  tchaton

  committed
  af865ff

  Copy full SHA for af865ff

Commits on Mar 19, 2021

• try something

  tchaton

  committed
  6147e6e

  Copy full SHA for 6147e6e

• update

  tchaton

  committed
  5053b4f

  Copy full SHA for 5053b4f

• add support for test and predict

  tchaton

  committed
  3ae2fbf

  Copy full SHA for 3ae2fbf

• update

  tchaton

  committed
  1d7ac88

  Copy full SHA for 1d7ac88

• update

  tchaton

  committed
  64f8a58

  Copy full SHA for 64f8a58

• use getattr

  tchaton

  committed
  655ff46

  Copy full SHA for 655ff46

• test

  tchaton

  committed
  3f5c4d6

  Copy full SHA for 3f5c4d6

• test

  tchaton

  committed
  ced0d66

  Copy full SHA for ced0d66

• update

  tchaton

  committed
  662f8de

  Copy full SHA for 662f8de

• tests

  tchaton

  committed
  753f63b

  Copy full SHA for 753f63b

• update

  tchaton

  committed
  e6a6917

  Copy full SHA for e6a6917

• update

  tchaton

  committed
  a7f7f4c

  Copy full SHA for a7f7f4c

• update

  tchaton

  committed
  c221d7d

  Copy full SHA for c221d7d

• update

  tchaton

  committed
  e39f6d7

  Copy full SHA for e39f6d7

• update

  tchaton

  committed
  e485850

  Copy full SHA for e485850

• remove file

  tchaton

  committed
  00d355a

  Copy full SHA for 00d355a

• update

  tchaton

  committed
  61da390

  Copy full SHA for 61da390

• Merge branch 'update-pytorch-profiler' of https://github.com/PyTorchLightning/pytorch-lightning into update-pytorch-profiler

  tchaton

  committed
  002f137

  Copy full SHA for 002f137

• update

  tchaton

  committed
  e565e8a

  Copy full SHA for e565e8a

• update

  tchaton

  committed
  6f2258f

  Copy full SHA for 6f2258f

Commits on Mar 21, 2021

• update

  tchaton

  committed
  883b242

  Copy full SHA for 883b242

• update

  tchaton

  committed
  9b957bd

  Copy full SHA for 9b957bd

• test

  tchaton

  committed
  70cc20d

  Copy full SHA for 70cc20d

• update#

  tchaton

  committed
  226e0be

  Copy full SHA for 226e0be

• update

  tchaton

  committed
  27348c7

  Copy full SHA for 27348c7

• update tests

  tchaton

  committed
  853d36c

  Copy full SHA for 853d36c

• update

  tchaton

  committed
  9004e92

  Copy full SHA for 9004e92

• add suport for 1.8

  tchaton

  committed
  c35e927

  Copy full SHA for c35e927

• rename records

  tchaton

  committed
  1f9d2d9

  Copy full SHA for 1f9d2d9

• add support for 1.8

  tchaton

  committed
  13caa86

  Copy full SHA for 13caa86

• update

  tchaton

  committed
  18c48ce

  Copy full SHA for 18c48ce

• resolve flake8

  tchaton

  committed
  a1c37ad

  Copy full SHA for a1c37ad

• resolve test

  tchaton

  committed
  3750b22

  Copy full SHA for 3750b22

Commits on Mar 22, 2021

• Merge branch 'master' into update-pytorch-profiler

  carmocca

  authored
  9a87c52

  Copy full SHA for 9a87c52

• Refactor basic profilers

  carmocca

  committed
  fbbb9a2

  Copy full SHA for fbbb9a2

• Fixes

  carmocca

  committed
  2a82e05

  Copy full SHA for 2a82e05

• Unused import

  carmocca

  committed
  ff125e2

  Copy full SHA for ff125e2

• Introduce setup

  carmocca

  committed
  01a760e

  Copy full SHA for 01a760e

• Profile on all ranks. Print to stdout on 0

  carmocca

  committed
  b31831e

  Copy full SHA for b31831e

• Introduce dirpath + filename

  carmocca

  committed
  f8a8772

  Copy full SHA for f8a8772

• CHANGELOG

  carmocca

  committed
  aa4b7dd

  Copy full SHA for aa4b7dd

• Add tests. Address comments

  carmocca

  committed
  8e3034e

  Copy full SHA for 8e3034e

• Merge branch 'update-pytorch-profiler' into update-pytorch-profiler-1.8

  tchaton

  authored
  96259a4

  Copy full SHA for 96259a4

• add on_run_stage_setup

  tchaton

  committed
  e4e0dd6

  Copy full SHA for e4e0dd6

• add on_run_stage_setup function

  tchaton

  committed
  d0fdbb9

  Copy full SHA for d0fdbb9

• update

  tchaton

  committed
  1a16bb3

  Copy full SHA for 1a16bb3

• add test for RegisterRecordFunction

  tchaton

  committed
  52fa69b

  Copy full SHA for 52fa69b

• update lightnng flow direction

  tchaton

  committed
  63b6988

  Copy full SHA for 63b6988

• move variable to private

  tchaton

  committed
  1cf5a64

  Copy full SHA for 1cf5a64

• remove trace

  tchaton

  committed
  a05acdd

  Copy full SHA for a05acdd

• Merge branch 'master' into refactor-base-profilers

  tchaton

  committed
  af72dff

  Copy full SHA for af72dff

• Undo code that should be in 3/4

  carmocca

  committed
  59c941b

  Copy full SHA for 59c941b

• Multi-stage multi-rank

  carmocca

  committed
  da0f310

  Copy full SHA for da0f310

• 2/5 changes

  carmocca

  committed
  59c1b4c

  Copy full SHA for 59c1b4c

• Pass stage in __del__

  carmocca

  committed
  dd1dce0

  Copy full SHA for dd1dce0

• Merge branch 'master' into refactor-base-profilers

  carmocca

  committed
  12d014b

  Copy full SHA for 12d014b

• Remove TODOs

  carmocca

  committed
  097a426

  Copy full SHA for 097a426

• Describe on_evaluation_end. Add tests

  carmocca

  committed
  4d529fa

  Copy full SHA for 4d529fa

• Typo

  carmocca

  committed
  58dcd4e

  Copy full SHA for 58dcd4e

• Address comments

  carmocca

  committed
  c37162f

  Copy full SHA for c37162f

• deepcopy tests

  carmocca

  committed
  4c5f1f3

  Copy full SHA for 4c5f1f3

• Advanced teardown

  carmocca

  committed
  5ed73fb

  Copy full SHA for 5ed73fb

• Fix teardown test

  carmocca

  committed
  897f8e5

  Copy full SHA for 897f8e5

• Fix tests

  carmocca

  committed
  e42be2a

  Copy full SHA for e42be2a

• Minor change

  carmocca

  committed
  32c301c

  Copy full SHA for 32c301c

• Update CHANGELOG.md

  carmocca

  authored
  af0c8ad

  Copy full SHA for af0c8ad

• Fix test

  carmocca

  committed
  29a73c5

  Copy full SHA for 29a73c5

• Merge branch 'refactor-base-profilers' into update-pytorch-profiler

  carmocca

  committed
  d6ede58

  Copy full SHA for d6ede58

• Quick fixes

  carmocca

  committed
  0dc1e06

  Copy full SHA for 0dc1e06

• Fix 6522

  carmocca

  committed
  cb756b8

  Copy full SHA for cb756b8

Commits on Mar 23, 2021

• resolve ddp tests

  tchaton

  committed
  758b942

  Copy full SHA for 758b942

• resolve tests

  tchaton

  committed
  fca4eb2

  Copy full SHA for fca4eb2

• resolve some tests

  tchaton

  committed
  2919a39

  Copy full SHA for 2919a39

• update tests

  tchaton

  committed
  d7ca5fa

  Copy full SHA for d7ca5fa

• Merge branch 'refactor-base-profilers' into update-pytorch-profiler

  tchaton

  committed
  34b7991

  Copy full SHA for 34b7991

• resolve tests

  tchaton

  committed
  b3ffe64

  Copy full SHA for b3ffe64

• Merge branch 'master' into update-pytorch-profiler

  tchaton

  committed
  e0d8308

  Copy full SHA for e0d8308

• update

  tchaton

  committed
  ecd1fd2

  Copy full SHA for ecd1fd2

• Merge branch 'update-pytorch-profiler-1.8' of https://github.com/PyTorchLightning/pytorch-lightning into update-pytorch-profiler-1.8

  tchaton

  committed
  3a0d821

  Copy full SHA for 3a0d821

• Merge branch 'update-pytorch-profiler' into update-pytorch-profiler-1.8

  tchaton

  committed
  c3e52f8

  Copy full SHA for c3e52f8

• resolve tests

  tchaton

  committed
  97c87d3

  Copy full SHA for 97c87d3

• resolve some tests

  tchaton

  committed
  172b388

  Copy full SHA for 172b388

• Missed fixes from 3/5

  carmocca

  committed
  ae31f00

  Copy full SHA for ae31f00

• Fixes

  carmocca

  committed
  78970d5

  Copy full SHA for 78970d5

• resolve some tests

  tchaton

  committed
  7e27a8d

  Copy full SHA for 7e27a8d

• Merge branch 'update-pytorch-profiler-1.8' of https://github.com/PyTorchLightning/pytorch-lightning into update-pytorch-profiler-1.8

  tchaton

  committed
  dbd1c7e

  Copy full SHA for dbd1c7e

• Merge branch 'update-pytorch-profiler' of https://github.com/PyTorchLightning/pytorch-lightning into update-pytorch-profiler-1.8

  tchaton

  committed
  c0107e2

  Copy full SHA for c0107e2

• resolve test for 1.7.1

  tchaton

  committed
  87aa09f

  Copy full SHA for 87aa09f

• Broken refactor

  carmocca

  committed
  ba067fd

  Copy full SHA for ba067fd

• Missed stage

  carmocca

  committed
  f916013

  Copy full SHA for f916013

• Minor changes

  carmocca

  committed
  9a1b1b6

  Copy full SHA for 9a1b1b6

• resolve tests

  tchaton

  committed
  04abd0a

  Copy full SHA for 04abd0a

• Merge branch 'update-pytorch-profiler' of https://github.com/PyTorchLightning/pytorch-lightning into update-pytorch-profiler

  tchaton

  committed
  ab1218e

  Copy full SHA for ab1218e

• Update CHANGELOG

  carmocca

  committed
  0230330

  Copy full SHA for 0230330

• resolve bug

  tchaton

  committed
  cba643b

  Copy full SHA for cba643b

• Merge branch 'update-pytorch-profiler' of https://github.com/PyTorchLightning/pytorch-lightning into update-pytorch-profiler

  tchaton

  committed
  85bfc7d

  Copy full SHA for 85bfc7d

• remove print

  tchaton

  committed
  af209f5

  Copy full SHA for af209f5

• Typo

  carmocca

  committed
  bb813bb

  Copy full SHA for bb813bb

• Cleanup

  carmocca

  committed
  624a85f

  Copy full SHA for 624a85f

• resolve ddp test

  tchaton

  committed
  1732ece

  Copy full SHA for 1732ece

• remove barrier

  tchaton

  committed
  f0c1675

  Copy full SHA for f0c1675

• update profiler

  tchaton

  committed
  fbe50b1

  Copy full SHA for fbe50b1

• Merge branch 'master' into update-pytorch-profiler

  tchaton

  authored
  a13664d

  Copy full SHA for a13664d

• update

  tchaton

  committed
  374f0da

  Copy full SHA for 374f0da

• Merge branch 'master' into update-pytorch-profiler-1.8

  tchaton

  authored
  e3c162e

  Copy full SHA for e3c162e

• Smaller model

  carmocca

  committed
  6e948ca

  Copy full SHA for 6e948ca

• update

  tchaton

  committed
  ad55628

  Copy full SHA for ad55628

• Merge branch 'update-pytorch-profiler-1.8' of https://github.com/PyTorchLightning/pytorch-lightning into update-pytorch-profiler-1.8

  tchaton

  committed
  05f68b4

  Copy full SHA for 05f68b4

• Merge branch 'update-pytorch-profiler' into update-pytorch-profiler-1.8

  tchaton

  committed
  88c44d6

  Copy full SHA for 88c44d6

• resolve tests

  tchaton

  committed
  2b0f92d

  Copy full SHA for 2b0f92d

• update

  tchaton

  committed
  e46cf1c

  Copy full SHA for e46cf1c

• Merge branch 'master' into update-pytorch-profiler-1.8

  tchaton

  committed
  5071947

  Copy full SHA for 5071947

• Minor changes. CHANGELOG

  carmocca

  committed
  5b51259

  Copy full SHA for 5b51259

• Minimize diff

  carmocca

  committed
  5aa19ec

  Copy full SHA for 5aa19ec

• update to 1.8.1

  tchaton

  committed
  efdb20b

  Copy full SHA for efdb20b

• Merge branch 'update-pytorch-profiler-1.8' of https://github.com/PyTorchLightning/pytorch-lightning into update-pytorch-profiler-1.8

  tchaton

  committed
  ea00c36

  Copy full SHA for ea00c36

• RunIf. Extra code. Check segfault

  carmocca

  committed
  79efcb2

  Copy full SHA for 79efcb2

• Merge branch 'update-pytorch-profiler-1.8' of https://github.com/PyTorchLightning/pytorch-lightning into update-pytorch-profiler-1.8

  carmocca

  committed
  5453bbe

  Copy full SHA for 5453bbe

• resolve tests

  tchaton

  committed
  11e3150

  Copy full SHA for 11e3150

• Typo. Bad merge

  carmocca

  committed
  2cd2be1

  Copy full SHA for 2cd2be1

• Merge branch 'update-pytorch-profiler-1.8' of https://github.com/PyTorchLightning/pytorch-lightning into update-pytorch-profiler-1.8

  tchaton

  committed
  6aab082

  Copy full SHA for 6aab082

• Merge branch 'update-pytorch-profiler-1.8' of https://github.com/PyTorchLightning/pytorch-lightning into update-pytorch-profiler-1.8

  tchaton

  committed
  7cff95a

  Copy full SHA for 7cff95a

• Fixing a bad merge

  carmocca

  committed
  496efab

  Copy full SHA for 496efab

• replace for kineto

  tchaton

  committed
  9ced644

  Copy full SHA for 9ced644

• Update pytorch_lightning/profiler/pytorch.py

  Show description for ee85864
  

  tchaton

  and

  ananthsub

  authored
  ee85864

  Copy full SHA for ee85864

• Update pytorch_lightning/profiler/pytorch.py

  Show description for efaa459
  

  tchaton

  and

  ananthsub

  authored
  efaa459

  Copy full SHA for efaa459

• Minor changes

  carmocca

  committed
  e73becc

  Copy full SHA for e73becc

• Merge branch 'update-pytorch-profiler-1.8' of https://github.com/PyTorchLightning/pytorch-lightning into update-pytorch-profiler-1.8

  carmocca

  committed
  2865c96

  Copy full SHA for 2865c96

• Bad merge

  carmocca

  committed
  92fa085

  Copy full SHA for 92fa085

• Use lists for flexibility

  carmocca

  committed
  a5093cb

  Copy full SHA for a5093cb

• Use sets

  carmocca

  committed
  5ccbfd0

  Copy full SHA for 5ccbfd0

• predict_step

  carmocca

  committed
  d1c0915

  Copy full SHA for d1c0915

• Ananth's suggestion

  carmocca

  committed
  8098aaf

  Copy full SHA for 8098aaf

• update

  tchaton

  committed
  8b9bbf0

  Copy full SHA for 8b9bbf0

• Docs

  carmocca

  committed
  a74e835

  Copy full SHA for a74e835

• Merge branch 'update-pytorch-profiler-1.8' of https://github.com/PyTorchLightning/pytorch-lightning into update-pytorch-profiler-1.8

  carmocca

  committed
  5af66be

  Copy full SHA for 5af66be

• Update pl_examples/basic_examples/profiler_example.py

  Show description for 3605b1a
  

  tchaton

  and

  Borda

  authored
  3605b1a

  Copy full SHA for 3605b1a

• update example

  tchaton

  committed
  8e93145

  Copy full SHA for 8e93145

• update example

  tchaton

  committed
  af5d2ef

  Copy full SHA for af5d2ef

• definition

  Borda

  committed
  f30fa58

  Copy full SHA for f30fa58

• yapf

  Borda

  committed
  9b5f90c

  Copy full SHA for 9b5f90c

• move pm

  Borda

  committed
  92eaf7d

  Copy full SHA for 92eaf7d

• typo

  Borda

  committed
  8a631e0

  Copy full SHA for 8a631e0

• update

  tchaton

  committed
  8f95814

  Copy full SHA for 8f95814

• Merge branch 'update-pytorch-profiler-1.8' of https://github.com/PyTorchLightning/pytorch-lightning into update-pytorch-profiler-1.8

  tchaton

  committed
  175f285

  Copy full SHA for 175f285