saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-30 23:18:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/onnxruntime/test/python/transformers
History
Ye Wang d433aa2459
Add transformers tool test to pipeline (#7959) 
* checkin transformers pipeline

* add docker requirements

* only trigger linux cpu

* temp remove tf instalation due to numpy version conflicts

* test numpy>=1.7

* revert numpy and disable transformers

* add coloredlogs

* enable shape_infer_helper and install transformers when needed

* pip3?

* testtest

* enable more tets

* line too long

* remove pytorch1.4 test and added back some onnx  files

* add tests

* copy dir

* disable 2 teests

* trim lines

* add missing onnx

* fix type

* fix  version conflicts

* install psutil

* change file path

* mfix path

* remove cached files

* add back attention fusion test

* labeled the shape infer test as slow

* fix

* enable tf2onnx test and enable pytest

* refactor path

* fix typo

* add cwd
2021-06-08 19:43:59 -07:00
..
test_data Add transformers tool test to pipeline (#7959) 2021-06-08 19:43:59 -07:00
bert_model_generator.py Add transformers tool test to pipeline (#7959) 2021-06-08 19:43:59 -07:00
conftest.py Add transformers tool test to pipeline (#7959) 2021-06-08 19:43:59 -07:00
test_attention_fusion.py Add transformers tool test to pipeline (#7959) 2021-06-08 19:43:59 -07:00
test_gelu_fusions.py Add transformers tool test to pipeline (#7959) 2021-06-08 19:43:59 -07:00
test_gpt2.py Add transformers tool test to pipeline (#7959) 2021-06-08 19:43:59 -07:00
test_optimizer.py Add transformers tool test to pipeline (#7959) 2021-06-08 19:43:59 -07:00
test_profiler.py Add transformers tool test to pipeline (#7959) 2021-06-08 19:43:59 -07:00
test_shape_infer_helper.py Add transformers tool test to pipeline (#7959) 2021-06-08 19:43:59 -07:00