Code

Support Apple MPS acceleration#1129

Code

Open

ClaudiaComito wants to merge 150 commits intomainhelmholtz-analytics/heat:mainfrom features/1053-support-Apple-silicon-GPUshelmholtz-analytics/heat:features/1053-support-Apple-silicon-GPUsCopy head branch name to clipboard

+2,446-1,773

Commits

Commits on Nov 26, 2022

• Support Apple MPS, first pass

  ClaudiaComito

  committed
  4e9ed3a

Commits on Mar 18, 2023

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  50297d7

• Include torch 2.0 in device check

  ClaudiaComito

  committed
  febdcfc

Commits on Mar 29, 2023

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  a5642e9

• reinstate quick_start.md

  ClaudiaComito

  committed
  8445476

• Merge branch 'docs/reinstate-quick-start' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  8763146

• [skip ci] edits

  ClaudiaComito

  committed
  92306e1

Commits on Apr 17, 2023

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  authored
  b0d7f0f

Commits on Apr 18, 2023

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  authored
  ffa014e

Commits on Apr 24, 2023

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  authored
  6015ebf

Commits on Apr 27, 2023

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  f3a5ad8

• fix tolerance for torch 2

  ClaudiaComito

  committed
  a96441b

Commits on May 18, 2023

• implement __array__ method

  ClaudiaComito

  committed
  c7b70c6

• test __array__ method

  ClaudiaComito

  committed
  ff8af94

• test __array__ method

  ClaudiaComito

  committed
  fc059c2

• Merge branch 'features/1153_array_method' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  ebf4c51

Commits on May 22, 2023

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  authored
  5c471e2

• Merge branch 'features/1053-support-Apple-silicon-GPUs' of github.com:helmholtz-analytics/heat into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  34c9347

Commits on Jun 16, 2023

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  f4e3afd

Commits on Jun 19, 2023

• Merge branch 'features/#1117-array-copy-None' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  681086f

• dtype changes for MPS device

  ClaudiaComito

  committed
  3ad2718

Commits on Jun 20, 2023

• accomodate single prec dtypes for MPS

  ClaudiaComito

  committed
  b50d9f6

• torch.linalg.inv workaround on MPS

  ClaudiaComito

  committed
  8ff410a

• Implement MPS-friendly choice of dtypes

  ClaudiaComito

  committed
  e3bd1d2

• skip hSVD tests on MPS, torch.norm unstable

  ClaudiaComito

  committed
  7564bac

• cast operands to single precision on MPS

  ClaudiaComito

  committed
  cd49de9

• do not cast to double precision on MPS

  ClaudiaComito

  committed
  3559295

• skip complex dtype tests on MPS

  ClaudiaComito

  committed
  cec0e75

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  2361713

Commits on Aug 13, 2023

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  3fb3667

Commits on Aug 16, 2023

• skip MPI op tests on Apple MPS

  ClaudiaComito

  committed
  7fbbc45

Commits on Oct 18, 2023

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  6f00be8

Commits on Nov 27, 2023

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  859b844

Commits on Apr 13, 2024

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  9414360

Commits on May 28, 2024

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  4cc7401

• test on float32 only on MPS

  ClaudiaComito

  committed
  e7a8c39

• skip test on MPS, ComplexFloat not supported

  ClaudiaComito

  committed
  6187804

• test float32 only on MPS

  ClaudiaComito

  committed
  0d464a3

• test float32 only on MPS

  ClaudiaComito

  committed
  b858a89

• edit docs

  ClaudiaComito

  committed
  c887a91

• non-distr in-place cumprod

  ClaudiaComito

  committed
  a58be6c

• no float64 on MPS

  ClaudiaComito

  committed
  2b71ed2

• skip float64 tests on MPS

  ClaudiaComito

  committed
  83d0136

• implement non-distr in-place cumsum

  ClaudiaComito

  committed
  a72678d

• catch MPS hypot crash on int64 early

  ClaudiaComito

  committed
  30cfa8a

• do not test non-supported dtypes on MPS

  ClaudiaComito

  committed
  9e128d6

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  95dcd0e

• replace device check with getattr to accomodate scalars

  ClaudiaComito

  committed
  cdef743

Commits on May 29, 2024

• skip complex-math tests on MPS and MacOS<14

  ClaudiaComito

  committed
  e6db9ae

• skip float64 tests on MPS

  ClaudiaComito

  committed
  00f6791

Commits on Jun 4, 2024

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  6c532f5

• do not test float64 on MPS

  ClaudiaComito

  committed
  8e7e2a6

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  6202f3b

Commits on Jun 6, 2024

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  113e562

• do not test float64 on MPS

  ClaudiaComito

  committed
  bf3bbb9

Commits on Jun 7, 2024

• skip float64 tests on MPS

  ClaudiaComito

  committed
  47b3253

Commits on Jun 12, 2024

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  3e8c8aa

• do not cast to float64 on MPS

  ClaudiaComito

  committed
  ad165fa

Commits on Jun 19, 2024

• do not test float64 and complex on mps

  ClaudiaComito

  committed
  d71c9c1

• early exit for non-distr flip

  ClaudiaComito

  committed
  84e4942

Commits on Jul 8, 2024

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  1b1d18d

• int32 cumsum on MPS

  ClaudiaComito

  committed
  2587b89

• skip reduce ops on 4+-dim arrays on MPS

  ClaudiaComito

  committed
  1db5cd6

• skip float64 tests on MPS

  ClaudiaComito

  committed
  7859f27

Commits on Jul 10, 2024

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  92fcafe

Commits on Jul 12, 2024

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  391d3e8

• skip int64 cumsum test on MPS

  ClaudiaComito

  committed
  34192f9

Commits on Jul 17, 2024

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  0085388

• indexing tests on MPS

  ClaudiaComito

  committed
  f596b73

Commits on Jul 18, 2024

• skip partitioned tests on MPS

  ClaudiaComito

  committed
  015ba20

Commits on Sep 9, 2024

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  4dda4ea

Commits on Sep 10, 2024

• add is_mps property

  ClaudiaComito

  committed
  0df738a

• refactor roll

  ClaudiaComito

  committed
  e096d33

• refactor tests

  ClaudiaComito

  committed
  32d20f2

• refactor tests

  ClaudiaComito

  committed
  52b1f14

Commits on Sep 12, 2024

• remove unnecessary try/except

  ClaudiaComito

  committed
  88740b9

• allow double precision int input

  ClaudiaComito

  committed
  79a7e44

• simplify MPS heuristics

  ClaudiaComito

  committed
  ead4ac3

• update tests

  ClaudiaComito

  committed
  9a71d7e

• cum ops now supported on int64

  ClaudiaComito

  committed
  a3a16c4

• update tests

  ClaudiaComito

  committed
  493c877

• remove ref to torch < 2

  ClaudiaComito

  committed
  8ca15dd

• allow cum ops on int64

  ClaudiaComito

  committed
  26c003d

• update tests

  ClaudiaComito

  committed
  cd23eaf

• update tests

  ClaudiaComito

  committed
  784d5a1

Commits on Sep 13, 2024

• cum ops now supported

  ClaudiaComito

  committed
  2330492

• update tests

  ClaudiaComito

  committed
  fc9b03a

• update tests

  ClaudiaComito

  committed
  06b2af7

Commits on Sep 30, 2024

• update tests

  ClaudiaComito

  committed
  c81300a

• update test_svd

  ClaudiaComito

  committed
  7b1fe5b

• update test_svdtools

  ClaudiaComito

  committed
  fb75a2e

• simplify is_mps

  ClaudiaComito

  committed
  da63ce8

• WIP - update test_statistics

  ClaudiaComito

  committed
  35cea1c

Commits on Oct 1, 2024

• bypass allreduce call for non-distr histc

  ClaudiaComito

  committed
  b23c6b7

• adapt test_statistics

  ClaudiaComito

  committed
  ae3a96d

• percentile output dtype on MPS

  ClaudiaComito

  committed
  d998e25

• adapt test_manipulations to MPS

  ClaudiaComito

  committed
  1a7ac24

• early out in non-distr cases

  ClaudiaComito

  committed
  bffd3db

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  22d6e0d

Commits on Oct 2, 2024

• update test_complex_math

  ClaudiaComito

  committed
  5628ea4

• update test_exp

  ClaudiaComito

  committed
  727cadd

• update test_factories

  ClaudiaComito

  committed
  aaff882

• update test_io

  ClaudiaComito

  committed
  8e44f89

• update test_logical

  ClaudiaComito

  committed
  386cd68

• skip tests on MPS

  ClaudiaComito

  committed
  8df2649

• update cumops for MPS

  ClaudiaComito

  committed
  550e638

• skip print_GPU on MPS

  ClaudiaComito

  committed
  00c6aab

Commits on Oct 9, 2024

• support device setting for randperm

  ClaudiaComito

  committed
  c31f584

Commits on Oct 10, 2024

• expand randperm docs on device

  ClaudiaComito

  committed
  6aec0b4

• adapt tests to MPS

  ClaudiaComito

  committed
  fcf7a9e

• adapt test_rounding to MPS

  ClaudiaComito

  committed
  7ca9548

• do not cast to float64 on MPS

  ClaudiaComito

  committed
  3c94c88

• adapt test_signal to MPS

  ClaudiaComito

  committed
  9d49269

Commits on Oct 11, 2024

• adatpt basic_tests to MPS

  ClaudiaComito

  committed
  827e231

Commits on Oct 14, 2024

• adapt test_trigonometrics to MPS

  ClaudiaComito

  committed
  222e8f0

Commits on Nov 11, 2024

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  authored
  e9d4eff

• adapt TestRSVD to MPS

  ClaudiaComito

  committed
  61daa63

• adapt test_randint to MPS

  ClaudiaComito

  committed
  f7db54e

• do not cast to float64 on MPS

  ClaudiaComito

  committed
  2e0b686

• adapt test_types to MPS

  ClaudiaComito

  committed
  21de13e

Commits on Nov 26, 2024

• skip MPS FFT tests on MacOS < 14

  ClaudiaComito

  committed
  509cb3b

• skip Threefry tests on MPS

  ClaudiaComito

  committed
  2bd212a

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  e9db00e

Commits on Nov 29, 2024

• adatpt test_fft to MPS

  ClaudiaComito

  committed
  10ec173

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  2993c11

• skip float64 test_isvd on MPS

  ClaudiaComito

  committed
  a20fab9

• adapt test_pca to MPS

  ClaudiaComito

  committed
  b3b72c5

• no sparse tests on MPS

  ClaudiaComito

  committed
  718b47e

• adapt to MPS

  ClaudiaComito

  committed
  6177f94

• adapt to MPS

  ClaudiaComito

  committed
  667f436

• remove print statement

  ClaudiaComito

  committed
  11d87b1

• skip float64 tests on MPS

  ClaudiaComito

  committed
  06db453

• skip complex128 on MPS

  ClaudiaComito

  committed
  a2cfa29

Commits on Dec 2, 2024

• fix CPU tests

  ClaudiaComito

  committed
  0875b5c

• fix CPU tests

  ClaudiaComito

  committed
  cd1b2b4

Commits on Dec 3, 2024

• update test_solver

  ClaudiaComito

  committed
  5147c7b

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  0589f93

• skip float64 batch-qr test on MPS

  ClaudiaComito

  committed
  0528481

Commits on Dec 10, 2024

• adapt test_permutation

  ClaudiaComito

  committed
  0efad5e

• Merge branch 'main' into features/1053-support-Apple-silicon-GPUs

  ClaudiaComito

  committed
  0a6b6ed

• return inv as DNDarray, not Tensor

  ClaudiaComito

  committed
  164b7b7

Commits on Dec 11, 2024

• skip DMD tests on MPS

  ClaudiaComito

  committed
  c7cdf34

• increase allclose tolerance for test_inv

  ClaudiaComito

  committed
  dc2ab84

• skip line formatting

  ClaudiaComito

  committed
  6907c71

• skip line formatting

  ClaudiaComito

  committed
  58cc44f

• skip line formatting

  ClaudiaComito

  committed
  1e902d1

Commits on Dec 12, 2024

• debugging test_sort on AMD

  ClaudiaComito

  committed
  71e83c5

• update test_iris

  ClaudiaComito

  committed
  fa3e900

• indices sorting workaround for CUDA

  ClaudiaComito

  committed
  103dc7e

• update PR template, docs

  ClaudiaComito

  committed
  80a867e