onnxruntime/tools/ci_build
Suffian Khan 7a3c1787af
Add CI pipeline to publish Python training package targeting Rocm (#7417)
* first attempt rocm training wheel

* modifications needed to python packaging pipeline for Rocm 4.1

* changges to not conflict with cuda

missed stage1 changes

remove package push

add option r to getopt

try again without python install

try again without python install

try again without python install

split pipelines and add back push to remote storage

try on cuda gpu pool

try again

try again

try running without az subscription set

try again on original pipeline

change pool

passing AMD Rocm whl on AMD-GPU pool

split rocm pipeline from cuda pipeline

remove comments

* try adding Rocm tests as well

* try with tests in place

* fix trailing ws

* add training data

* try again as root for tests

* use python3

* typo

* try to map video, render group into container

* try again

* try again

* try to avoid yum error code

* make UID 1001

* try without yum downgrade

* define rocm_version=None

* remove CUDA related comments for Rocm Dockerfile

* Dont pin nightly torch torchvision torchtext versions as they expire (for now nightly is required for Rocm 4.1)

* missed requirements-rocm.txt from last commit

* fix whitespace
2021-04-23 17:22:31 -07:00
..
github Add CI pipeline to publish Python training package targeting Rocm (#7417) 2021-04-23 17:22:31 -07:00
wasm build ONNXRuntime into WebAssembly (#6478) 2021-04-06 16:18:10 -07:00
__init__.py
amd_hipify.py Polish Adam kernel (#7294) 2021-04-09 01:11:09 -07:00
build.py Add CI pipeline to publish Python training package targeting Rocm (#7417) 2021-04-23 17:22:31 -07:00
clean_docker_image_cache.py Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
coverage.py Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
gen_def.py
get_docker_image.py Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
logger.py
op_registration_utils.py Support required types when excluding typed registrations (#6871) 2021-03-08 08:22:07 -08:00
op_registration_validator.py Support specifying globally allowed types from build script (#6677) 2021-02-22 14:05:00 -08:00
reduce_op_kernels.py Support required types when excluding typed registrations (#6871) 2021-03-08 08:22:07 -08:00
upload_python_package_to_azure_storage.py for training to use new python package naming convention to explicitl… (#7204) 2021-04-13 16:19:42 -07:00