onnxruntime/tools/ci_build
Chi Lo 7242627fec
Integrate TensorRT into GPU Python package (#9785)
* add use_tensorrt build option

* Add use_tensorrt to running tests

* add use_tensorrt for Windows

* make trt ep to skip backend test

* make trt ep to skip backend test

* Fix bug

* Add/Modify description

* modify for debug

* swtich pool to test

* modify to debug

* modify to debug

* add vobersity

* refine the code

* refine the code

* refine the code

* fix flake8 warning

* refine the code

* add pre_load check for trt as well as add cupti lib to cuda depedencies

* modify script to make trt build path the same as cuda

* show error message when user wants to run TensorRT but TensorRT is not installed in the env

* fix bug

* fix bug

* add trt lib for manylinux

* include cuda_dependencies for trt

* rewrite the condition to throw exception

* make code more compact
2021-11-18 13:26:51 -08:00
..
github Integrate TensorRT into GPU Python package (#9785) 2021-11-18 13:26:51 -08:00
__init__.py Add validation of op registrations (#5817) 2020-11-17 10:44:09 -08:00
amd_hipify.py Implement block wise softmax for reduction dimention > 1024 cases. (#9696) 2021-11-14 11:47:58 +08:00
build.py Integrate TensorRT into GPU Python package (#9785) 2021-11-18 13:26:51 -08: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 Minimize changes to fix missing symbols used from C# (#8867) 2021-08-28 07:10:14 +10:00
get_docker_image.py Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
logger.py Cache build docker images in container registry. (#5811) 2020-11-17 17:02:24 -08:00
op_registration_utils.py Remove Featurizers code (#9300) 2021-10-20 10:20:35 -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 Restore transformers tests and disable some tests (#8530) 2021-07-29 14:09:36 -07:00
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