saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-11 17:48:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/onnxruntime/test/python
History
Tianlei Wu 237076a660
Add option to disable FastGelu half2 cuda kernel (#8819) 
Allow FastGelu half2 kernel to build without --cmake_extra_defines CMAKE_CUDA_ARCHITECTURES=xx
Add environment variable ORT_TRANSFORMER_OPTIONS=4 to disable half2 FastGelu kernel for testing purpose
Test parity of FastGelu operator with fp16 inputs.
2021-08-25 08:37:41 -07:00
..
contrib_ops Adding TorchEmbedding contrib op (#7136) 2021-03-30 14:33:25 -07:00
quantization Implement SparseTensor and infrastructure suppport and advance ONNX commit (#8038) 2021-07-22 15:24:36 -07:00
testdata Adding TorchEmbedding contrib op (#7136) 2021-03-30 14:33:25 -07:00
transformers Add option to disable FastGelu half2 cuda kernel (#8819) 2021-08-25 08:37:41 -07:00
helper.py
onnx_backend_test_series.py Update migraphx to rocm4.2 (#7994) 2021-06-22 13:39:51 -07:00
onnxruntime_test_ort_trainer.py Merge pull request #6973 from microsoft/thiagofc/merge-ortmodule-into-master 2021-03-17 10:30:06 -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 [eager mode] fix build and support customize shared provider entry point (#8680) 2021-08-11 15:10:35 -07:00
onnxruntime_test_python_backend.py Remove unused imports from Python tests. (#5405) 2020-10-08 17:24:10 -07: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 Remove unused imports from Python tests. (#5405) 2020-10-08 17:24:10 -07: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 Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
onnxruntime_test_python_nuphar.py [Nuphar] Add support for opset 14 (#8483) 2021-07-27 06:13:47 -07:00
onnxruntime_test_python_sparse_matmul.py Implement SparseTensor and infrastructure suppport and advance ONNX commit (#8038) 2021-07-22 15:24:36 -07:00
onnxruntime_test_python_symbolic_shape_infer.py Fix symbolic shape inference regression in RoBERTa training (#8364) 2021-07-13 08:15:53 -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