lightning

Commit Graph

Author	SHA1	Message	Date
awaelchli	a41528c2a6	Update tests for PyTorch 2.2.1 (#19521 )	2024-02-23 13:11:34 -05:00
Jirka Borovec	3bd133b107	CI: enable testing with coming PT 2.2 (#19289 ) * ci: build dockers for PT 2.2 * py3.12 * --pre --extra-index-url * typing-extensions * bump jsonargparse * install latest jsonargparse * Add windows skips for Fabric * convert to xfail * add pytorch skips * skip checkpoint consolidation test * set max torch --------- Co-authored-by: awaelchli <aedu.waelchli@gmail.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>	2024-01-26 16:42:09 +01:00
awaelchli	7cc79fe7ba	Reapply `torch.compile` in Fabric.setup() (#19280 )	2024-01-23 21:17:41 -05:00
awaelchli	d10e918ce0	Rewrite gradient clipping tests (#19262 )	2024-01-10 12:39:56 -05:00
Jirka Borovec	830a62a722	ruff: replace isort with ruff +TPU (#17684 ) * ruff: replace isort with ruff * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * fixing & imports * lines in warning test * docs * fix enum import * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * fixing * import * fix lines * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * type ClusterEnvironment * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>	2023-09-26 11:54:55 -04:00
Adrian Wälchli	b14ddd9c49	Fix state dict loading for ddp/dp in Fabric (#17997 ) * fix state dict loading for ddp/dp * test * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * changelog * update test * move params to same device before equality test * test strategy --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>	2023-07-06 13:47:17 +02:00

6 Commits