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
Viswanath Boga 85874bb315
embed layer fusion gpt2 (#9336) 
* Changes to fuse embed layer for gpt2, kernal changes pending

* verified add output and regular add match

* Test added for additional output embedlayernorm, working on CUDA

* Test passing on CPU

* updated convert_to_onnx toll to check parity correctly

* removed some debugs

* couple of TODO left as in optimizer.py

* removed changes to optimizer.py

* fixing build

* fixing build

* updated order of initilization

* added a test case for float16

* updating the docs

* updating tests failing due to embed layer fusion

* update unit tests

* updating CUDA documentation in operatorkernels.md

* addressing comments

* OperatorKernels.md updated with CUDA

* adding TODO to qembed_layer

* minor edit

* updated docs

* addressing comments

* adding position ids to embed layer gpt2

* updating fused gpt2 model

* added extra test

* remove comments

* addressing comments

* contrib_defs.cc updated

* all tests passing

* fixing a typo

* minor edit

* trigger build

* qembedlayernorm checkinputs updated

* fixing build error

* fixing build error

* fixing build error
2021-10-28 11:06:26 -07: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 embed layer fusion gpt2 (#9336) 2021-10-28 11:06:26 -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 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 Execution Provider Profiler (#8406) 2021-09-28 13:59:52 -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 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 Include pytorch_export_contrib_ops in inference builds (#8878) 2021-09-02 14:26:58 -07:00