ROCm / pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration
http://pytorch.org
Other
219 stars 51 forks source link

[release/2.3] Update apex commit to pick up wheel-related changes #1440

Closed jithunnair-amd closed 3 months ago

jithunnair-amd commented 3 months ago

Tested using http://rocm-ci.amd.com/job/rocm-pytorch-manylinux-wheel-builder/427 Artifacts generated: http://rocm-ci.amd.com/job/rocm-pytorch-manylinux-wheel-builder/427/artifact/result.cp39/*view*/ apex-1.3.0+rocm6.2.0-cp39-cp39-linux_x86_64.whl pushed to http://compute-artifactory.amd.com/artifactory/compute-pytorch-rocm/compute-rocm-rel-6.2/4/apex-1.3.0+rocm6.2.0-cp39-cp39-linux_x86_64.whl

L0 tests ran fine with: bash run_rocm.sh 2>&1 | tee run_rocm.log