onnxruntime/onnxruntime/test/python
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
..
contrib_ops Adding TorchEmbedding contrib op (#7136) 2021-03-30 14:33:25 -07:00
quantization Add QDQ for output of node (#9134) 2021-09-27 16:55:23 -07:00
testdata Adding TorchEmbedding contrib op (#7136) 2021-03-30 14:33:25 -07:00
transformers [python api] align api with other language bindings' treatment of explicit provider registrations. enforce use of providers param in python InferenceSession when execution providers other than default CPU are enabled. (#9712) 2021-11-10 12:17:53 -08:00
helper.py
onnx_backend_test_series.py Integrate TensorRT into GPU Python package (#9785) 2021-11-18 13:26:51 -08:00
onnxruntime_test_ort_trainer.py Remove previous temporary fixes and address TODOs (#9020) 2021-09-13 10:10:07 -07:00
onnxruntime_test_ort_trainer_with_mixed_precision.py Remove unused imports from Python tests. (#5405) 2020-10-08 17:24:10 -07:00
onnxruntime_test_python.py [python api] align api with other language bindings' treatment of explicit provider registrations. enforce use of providers param in python InferenceSession when execution providers other than default CPU are enabled. (#9712) 2021-11-10 12:17:53 -08:00
onnxruntime_test_python_backend.py [python api] align api with other language bindings' treatment of explicit provider registrations. enforce use of providers param in python InferenceSession when execution providers other than default CPU are enabled. (#9712) 2021-11-10 12:17:53 -08:00
onnxruntime_test_python_backend_mlops.py Remove unused imports from Python tests. (#5405) 2020-10-08 17:24:10 -07:00
onnxruntime_test_python_iobinding.py [python api] align api with other language bindings' treatment of explicit provider registrations. enforce use of providers param in python InferenceSession when execution providers other than default CPU are enabled. (#9712) 2021-11-10 12:17:53 -08:00
onnxruntime_test_python_keras.py Remove unused imports from Python tests. (#5405) 2020-10-08 17:24:10 -07:00
onnxruntime_test_python_mlops.py [python api] align api with other language bindings' treatment of explicit provider registrations. enforce use of providers param in python InferenceSession when execution providers other than default CPU are enabled. (#9712) 2021-11-10 12:17:53 -08:00
onnxruntime_test_python_nuphar.py [python api] align api with other language bindings' treatment of explicit provider registrations. enforce use of providers param in python InferenceSession when execution providers other than default CPU are enabled. (#9712) 2021-11-10 12:17:53 -08:00
onnxruntime_test_python_sparse_matmul.py [python api] align api with other language bindings' treatment of explicit provider registrations. enforce use of providers param in python InferenceSession when execution providers other than default CPU are enabled. (#9712) 2021-11-10 12:17:53 -08:00
onnxruntime_test_python_symbolic_shape_infer.py Fix shape inference issue in Gather op (#9147) 2021-09-28 22:46:12 -07:00
onnxruntime_test_training_unit_tests.py Remove unused imports from Python tests. (#5405) 2020-10-08 17:24:10 -07:00
onnxruntime_test_training_unittest_utils.py Remove unused imports from Python tests. (#5405) 2020-10-08 17:24:10 -07:00
test_pytorch_export_contrib_ops.py [python api] align api with other language bindings' treatment of explicit provider registrations. enforce use of providers param in python InferenceSession when execution providers other than default CPU are enabled. (#9712) 2021-11-10 12:17:53 -08:00