onnxruntime/orttraining/tools/amdgpu
Weixing Zhang 840212e115
Enable OneHot kernel for ROCm EP and add Dockerfile for ROCm 4.3.1 (#9656)
* enable OneHot for ROCm EP

* add dockerfile for ROCm 4.3.1

Co-authored-by: Weixing Zhang <wezhan@microsoft.com>
2021-12-07 12:47:00 -08:00
..
model
script Add GPU kernels for ROCm EP (#5655) 2020-11-06 16:11:06 -08:00
Dockerfile.rocm3.7 add dockerfile for ROCm3.10 and update BUILD.md for ROCm EP (#5821) 2020-12-08 23:14:56 -08:00
Dockerfile.rocm3.7.pytorch add dockerfile for ROCm3.10 and update BUILD.md for ROCm EP (#5821) 2020-12-08 23:14:56 -08:00
Dockerfile.rocm3.8.pytorch add dockerfile for ROCm3.10 and update BUILD.md for ROCm EP (#5821) 2020-12-08 23:14:56 -08:00
Dockerfile.rocm3.9 add dockerfile for ROCm3.10 and update BUILD.md for ROCm EP (#5821) 2020-12-08 23:14:56 -08:00
Dockerfile.rocm3.9.pytorch add dockerfile for ROCm3.10 and update BUILD.md for ROCm EP (#5821) 2020-12-08 23:14:56 -08:00
Dockerfile.rocm3.10.pytorch add dockerfile for ROCm3.10 and update BUILD.md for ROCm EP (#5821) 2020-12-08 23:14:56 -08:00
Dockerfile.rocm4.0.pytorch update Dockerfile for workaround for issue in RCCL for rocm4.0 (#7108) 2021-03-23 13:36:04 -07:00
Dockerfile.rocm4.1.pytorch [ROCm] dockerfile updates (#7955) 2021-06-10 23:50:19 -07:00
Dockerfile.rocm4.2.pytorch [ROCm] dockerfile updates (#7955) 2021-06-10 23:50:19 -07:00
Dockerfile.rocm4.3.1.pytorch Enable OneHot kernel for ROCm EP and add Dockerfile for ROCm 4.3.1 (#9656) 2021-12-07 12:47:00 -08:00