Code

Add AMDGPU extension#2189

Code

Merged

ToucheSir merged 16 commits intoFluxML:masterFluxML/Flux.jl:masterfrom pxl-th:pxl-th/amdgpuCopy head branch name to clipboardFeb 28, 2023

+513-81

Commits

Commits on Feb 14, 2023

• Initial implementation of AMDGPU extension

  pxl-th

  committed
  9799053

Commits on Feb 15, 2023

• Add more tests

  pxl-th

  committed
  30f076e

• Partially handle convolution

  pxl-th

  committed
  211ab21

Commits on Feb 16, 2023

• Handle convolutions correctly

  pxl-th

  committed
  d0eb6a0

• Add batchnorm

  pxl-th

  committed
  de91f9a

• Add gpu backend switch mechanism

  pxl-th

  committed
  a37ee90

• Update .gitignore

  pxl-th

  committed
  eaada4b

• Add AMDGPU to extras

  pxl-th

  committed
  37ce734

Commits on Feb 17, 2023

• Fix batchnorm & handle regular convolutions

  pxl-th

  committed
  cf6dd42

• Cleanup

  pxl-th

  committed
  0a9daf7

Commits on Feb 21, 2023

• Update NNlib compat

  pxl-th

  committed
  f68e54d

• Add documentation about AMD GPU support

  pxl-th

  committed
  1e8bfb2

Commits on Feb 27, 2023

• Handle ConvTranspose correctly & refactor

  pxl-th

  committed
  746caa5

• Merge remote-tracking branch 'origin/pxl-th/amdgpu' into pxl-th/amdgpu

  pxl-th

  committed
  98ce4e9

• Unify gpu test utils

  pxl-th

  committed
  54c4946

Commits on Feb 28, 2023

• Refactor

  pxl-th

  committed
  621829b