Flux.jl
Add AMDGPU extension
#2189
Merged
Overview
Commits
16
Changes
View On GitHub

Add AMDGPU extension #2189

ToucheSir merged 16 commits into FluxML:master from pxl-th/amdgpu
pxl-th
pxl-th Initial implementation of AMDGPU extension
97990530
pxl-th Add more tests
30f076e2
pxl-th Partially handle convolution
211ab212
pxl-th Handle convolutions correctly
d0eb6a03
pxl-th Add batchnorm
de91f9a7
pxl-th Add gpu backend switch mechanism
a37ee90b
pxl-th Update .gitignore
eaada4b5
pxl-th Add AMDGPU to extras
37ce734c
ToucheSir
ToucheSir commented on 2023-02-17
pxl-th Fix batchnorm & handle regular convolutions
cf6dd42d
pxl-th Cleanup
0a9daf72
pxl-th Update NNlib compat
f68e54d8
pxl-th pxl-th marked this pull request as ready for review 3 years ago
pxl-th Add documentation about AMD GPU support
1e8bfb2f
pxl-th
ToucheSir
ToucheSir commented on 2023-02-25
pxl-th Handle ConvTranspose correctly & refactor
746caa51
pxl-th Merge remote-tracking branch 'origin/pxl-th/amdgpu' into pxl-th/amdgpu
98ce4e9f
pxl-th Unify gpu test utils
54c49469
ToucheSir
ToucheSir commented on 2023-02-28
pxl-th Refactor
621829be
pxl-th
ToucheSir
ToucheSir approved these changes on 2023-02-28
ToucheSir ToucheSir merged cc1ae6c3 into master 3 years ago
pxl-th pxl-th deleted the pxl-th/amdgpu branch 3 years ago
codecov

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum