onnxruntime/tools/ci_build
Xinya Zhang eb827bd3e5
[ROCm] NGramRepeatBlock, LongformerAttention and DecoderAttention Ops (#11971)
* [ROCm] enable NGramRepeatBlock Op

* [ROCm] Enable testing ROCm in NGramRepeatBlockTest.NGramSize_3

Also link onnxruntime_test_all with amdhip64 when USE_ROCM=1

* [ROCm] add LongformerAttention Op

* [ROCm] Enable LongformerAttentionTest

* [ROCm] Add DecoderAttention Op

* Enable DecoderAttention Test for ROCm.

* [ROCM] Updates according to reviews
2022-08-11 19:32:08 -07:00
..
github Remove CUDA 10.2 support (#12541) 2022-08-10 22:46:41 -07:00
__init__.py
amd_hipify.py [ROCm] NGramRepeatBlock, LongformerAttention and DecoderAttention Ops (#11971) 2022-08-11 19:32:08 -07:00
build.py [js] loosen test timeout (#12427) 2022-08-02 19:01:19 -07:00
clean_docker_image_cache.py
coverage.py
gen_def.py Snpe ep (#11665) 2022-06-03 14:10:02 -07:00
get_docker_image.py Delete the build scripts that were copied from manylinux project (#12358) 2022-07-29 18:24:19 -07:00
logger.py
op_registration_utils.py
op_registration_validator.py
policheck_exclusions.xml
reduce_op_kernels.py
requirements.txt
upload_python_package_to_azure_storage.py