onnxruntime/tools/ci_build
Tang, Cheng 0adeb86bfd
Fix ortmodule for the pytorch model with ort device (#9927)
* add ortmodule and eager mode test

* add ortmodule dependency

* convert between aten ort tensor and ortvalue

* register the EP to ortmodule using ort device information

* remove duplicated test

* remove useless dependency

* handle half precision type for ortmodule outputs

* adjust the tensor conversion python code

Co-authored-by: Cheng Tang <chenta@microsoft.com@orttrainingdev9.d32nl1ml4oruzj4qz3bqlggovf.px.internal.cloudapp.net>
2021-12-08 13:04:29 -08:00
..
github Fix ortmodule for the pytorch model with ort device (#9927) 2021-12-08 13:04:29 -08:00
__init__.py
amd_hipify.py Decoder Attention CUDA Op (#9792) 2021-11-19 19:25:36 -08:00
build.py Enable proper override using MIMalloc (#9944) 2021-12-07 17:56:58 -08:00
clean_docker_image_cache.py
coverage.py
gen_def.py
get_docker_image.py
logger.py
op_registration_utils.py Remove Featurizers code (#9300) 2021-10-20 10:20:35 -07:00
op_registration_validator.py
reduce_op_kernels.py Improve reduced ops and types build (#9908) 2021-12-07 13:02:05 -08:00
requirements.txt
upload_python_package_to_azure_storage.py do not add pkg wheel entry to the index html file if it already exists (#9004) 2021-09-09 16:20:19 -07:00