Code

MPS Accelerator#13123

Code

Merged

lexierule merged 101 commits intomasterLightning-AI/pytorch-lightning:masterfrom mps_acceleratorLightning-AI/pytorch-lightning:mps_acceleratorCopy head branch name to clipboardJun 24, 2022

+812-163

Commits

Commits on May 21, 2022

• update accelerator and device parsing

  justusschock

  committed
  b1aea3d

• update runif checks

  justusschock

  committed
  17523f3

• update runif usages

  justusschock

  committed
  1ed3895

• more occurences of min_gpus

  justusschock

  authored
  020e036

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

  Show description for 13da4d0
  pre-commit-ci[bot]

  committed
  13da4d0

• fix mps accelerator

  justusschock

  committed
  aaa5967

• fix imports and device parser

  justusschock

  committed
  06ff778

• trainer integration

  justusschock

  committed
  7a2dd98

• docs

  justusschock

  committed
  f27a7cf

• fix runif

  justusschock

  committed
  4460caf

• update mps tests

  justusschock

  committed
  4787724

• update accelerator connector to reflect mps changes

  justusschock

  committed
  46eaa7c

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

  Show description for 1fe3a94
  pre-commit-ci[bot]

  committed
  1fe3a94

Commits on May 22, 2022

• add import

  justusschock

  committed
  2a08c61

• fix accelerator registry test

  justusschock

  committed
  1458acc

• fix some gpu tests

  justusschock

  committed
  b34c447

Commits on May 23, 2022

• fix gpu intput normalization

  justusschock

  committed
  3f17012

• adjust runif for mps to also allow requiring that it is not available

  justusschock

  committed
  b89efac

• separate auto choice tests for mps and gpu

  justusschock

  committed
  1fa485e

• pep8

  justusschock

  committed
  934b46a

• remove unnecessary block in docs

  justusschock

  committed
  70003d6

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

  Show description for b5ba668
  

  pre-commit-ci[bot]

  authored and

  justusschock

  committed
  b5ba668

• update docs

  justusschock

  committed
  6b00f0f

• docs

  

  Borda

  authored and

  justusschock

  committed
  e406b97

• mypy

  justusschock

  committed
  c41d059

• change wording from mps to apple silicon

  justusschock

  authored
  5a95095

• update mps tests

  justusschock

  committed
  56d5181

• update device parser

  justusschock

  committed
  6f263d6

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

  Show description for 0f832e0
  pre-commit-ci[bot]

  committed
  0f832e0

• Update pytorch_lightning/accelerators/mps.py

  Show description for ccfac24
  

  justusschock

  and

  carmocca

  authored
  ccfac24

• Update pytorch_lightning/utilities/device_parser.py

  Show description for 064f3af
  

  justusschock

  and

  carmocca

  authored
  064f3af

Commits on May 24, 2022

• update callback tests

  justusschock

  committed
  b93d1e4

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

  Show description for a1c71e1
  

  pre-commit-ci[bot]

  authored and

  justusschock

  committed
  a1c71e1

• update core

  justusschock

  committed
  ea340c9

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

  Show description for b88ec78
  

  pre-commit-ci[bot]

  authored and

  justusschock

  committed
  b88ec78

• upate lite tests

  justusschock

  committed
  2e995b5

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

  Show description for e643dee
  

  pre-commit-ci[bot]

  authored and

  justusschock

  committed
  e643dee

• update loops

  justusschock

  committed
  4ffc08e

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

  Show description for deeb60b
  

  pre-commit-ci[bot]

  authored and

  justusschock

  committed
  deeb60b

• models

  justusschock

  committed
  936fc0f

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

  Show description for 477377d
  

  pre-commit-ci[bot]

  authored and

  justusschock

  committed
  477377d

• strategies

  justusschock

  committed
  98ada0b

• trainer

  justusschock

  committed
  c48351c

• utilities

  justusschock

  committed
  394799e

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

  Show description for 16eb921
  

  pre-commit-ci[bot]

  authored and

  justusschock

  committed
  16eb921

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

  Show description for 3a93dc2
  pre-commit-ci[bot]

  committed
  3a93dc2

• fix pre-commit

  justusschock

  committed
  f7e2005

• fix typo

  justusschock

  authored
  6516826

• fix accelerator devices to always contain index

  justusschock

  committed
  11d7a2a

• add mps accelerator type for lite

  justusschock

  committed
  6f89c3e

• add accelerator type mps

  justusschock

  committed
  bc6da48

• fix callback tests

  justusschock

  committed
  3df39d2

• fix core tests

  justusschock

  committed
  12a22c4

• fix lite tests

  justusschock

  committed
  c594c83

• fix loop tests

  justusschock

  committed
  f705cc9

• fix model tests

  justusschock

  committed
  0d9f758

• fix strategy tests

  justusschock

  committed
  18c9e23

• fix logging tests

  justusschock

  committed
  0dac2d8

• fix trainer tests

  justusschock

  committed
  62190c3

• fix utility tests

  justusschock

  committed
  0afeb46

• Merge branch 'master' into mps_accelerator

  justusschock

  authored
  638d137

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

  Show description for 5e6c7ba
  pre-commit-ci[bot]

  committed
  5e6c7ba

• make all mps device tests to accept string and lazily construct devices to avoid failure due to unknown device type

  justusschock

  authored
  99ee709

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

  Show description for b37056e
  pre-commit-ci[bot]

  committed
  b37056e

• stupid bug

  justusschock

  authored
  dc151c5

• handle different torch versions

  justusschock

  authored
  62936c3

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

  Show description for 49ab319
  pre-commit-ci[bot]

  committed
  49ab319

• time to get some sleep

  justusschock

  authored
  c76fd40

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

  Show description for b791e47
  pre-commit-ci[bot]

  committed
  b791e47

• remove foreach

  justusschock

  authored
  d64cbf9

• make cpu tests pass (hopefully)

  justusschock

  authored
  ee0000b

• remove leftover breakpoint

  justusschock

  authored
  8e16631

Commits on May 25, 2022

• remove lefvtover from debugging

  justusschock

  committed
  0028737

• move _MPS_AVAILABLE to accelerator file

  justusschock

  committed
  31a60b3

• update printing logic trainer

  justusschock

  committed
  3e56f1f

• add warning

  justusschock

  committed
  e8e42ad

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

  Show description for 3f77e57
  pre-commit-ci[bot]

  committed
  3f77e57

• pep8

  justusschock

  committed
  85031f8

• Add changelog entry

  justusschock

  authored
  ef6fa81

• Merge branch 'master' into mps_accelerator

  justusschock

  authored
  2ae9b8c

• Update pytorch_lightning/trainer/trainer.py

  Show description for eec7ddb
  

  justusschock

  and

  carmocca

  authored
  eec7ddb

• Update tests/helpers/runif.py

  Show description for d21ed47
  

  justusschock

  and

  carmocca

  authored
  d21ed47

• adress comments from @carmocca

  justusschock

  authored
  578e4eb

• mypy

  justusschock

  committed
  6f75a8b

• Update pytorch_lightning/utilities/device_parser.py

  Show description for 45aa2e0
  

  justusschock

  and

  akihironitta

  authored
  45aa2e0

• Update tests/plugins/test_amp_plugins.py

  Show description for 717a53c
  

  justusschock

  and

  akihironitta

  authored
  717a53c

• add link

  justusschock

  authored
  cf6e7b0

• Update pytorch_lightning/accelerators/mps.py

  Show description for e44265f
  

  justusschock

  and

  akihironitta

  authored
  e44265f

Commits on May 31, 2022

• Update pytorch_lightning/accelerators/mps.py

  Show description for e74af6f
  

  justusschock

  and

  awaelchli

  authored
  e74af6f

• Update tests/accelerators/test_mps.py

  Show description for 3b252d7
  

  justusschock

  and

  awaelchli

  authored
  3b252d7

Commits on Jun 7, 2022

• comments from @awaelchi

  justusschock

  authored
  beaf933

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

  Show description for 5c364dd
  pre-commit-ci[bot]

  committed
  5c364dd

• fix wrong indent

  justusschock

  authored
  7af2f87

Commits on Jun 23, 2022

• resolve merge conflicts

  justusschock

  committed
  858708d

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

  Show description for 25d148b
  pre-commit-ci[bot]

  committed
  25d148b

• fix imports

  justusschock

  committed
  d267d9c

• fix test imports

  justusschock

  committed
  7e9c9f4

• fix imports

  justusschock

  committed
  e072d92

• fix pep8

  justusschock

  committed
  5b5a58e

Commits on Jun 24, 2022

• Apply suggestions from code review

  Show description for c783a6a
  

  justusschock

  and

  RobertLaurella

  authored
  c783a6a

• Update docs/source-pytorch/accelerators/mps_basic.rst

  Show description for 92e00b2
  

  justusschock

  and

  RobertLaurella

  authored
  92e00b2