onnxruntime/tools/ci_build
Thiago Crepaldi 83be3759bc
Add post-install command to build PyTorch CPP extensions from within onnxruntime package (#8027)
ORTModule requires two PyTorch CPP extensions that are currently JIT compiled. The runtime compilation can cause issues in some environments without all build requirements or in environments with multiple instances of ORTModule running in parallel

This PR creates a custom command to compile such extensions that must be manually executed before ORTModule is executed for the first time. When users try to use ORTModule before the extensions are compiled, an error with instructions are raised

PyTorch CPP Extensions for ORTModule can be compiled by running:
python -m onnxruntime.training.ortmodule.torch_cpp_extensions.install

Full build environment is needed for this
2021-06-28 18:11:58 -07:00
..
github Add post-install command to build PyTorch CPP extensions from within onnxruntime package (#8027) 2021-06-28 18:11:58 -07:00
wasm [wasm] emsdk: allow to install emscripten only (#7961) 2021-06-07 09:45:02 -07:00
__init__.py
amd_hipify.py Explicitly pass cuda stream to thrust function rather than use cuda default stream implicitly (#7414) 2021-04-25 01:18:56 -07:00
build.py Update C++ Standard from 14 to 17 (#8041) 2021-06-25 14:08:01 -07:00
clean_docker_image_cache.py
coverage.py
gen_def.py
get_docker_image.py
logger.py
op_registration_utils.py Update Android AAR packaging pipeline script (#7559) 2021-05-04 11:13:33 -07: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
requirements.txt Add post-install command to build PyTorch CPP extensions from within onnxruntime package (#8027) 2021-06-28 18:11:58 -07:00
upload_python_package_to_azure_storage.py to support multiple cuda versions in published onnxruntime-training package (#7468) 2021-04-27 17:15:33 -07:00