onnxruntime/tools/ci_build
ytaous d632f9a3fa
[ROCm] Enable Sampling Op UT on AMD (#14581)
Making basic porting effort to run Sampling UT on ROCm ep, based on the
commits:

https://github.com/microsoft/onnxruntime/pull/13426
https://github.com/microsoft/onnxruntime/pull/14218

1. enabling EmbedLayerNorm op
2. enabling Sampling op
3. enabling helpers to copy data from CPU->GPU for subgraph

This task is the first checkpoint. There could be other missing ops when
testing a real model.
We will migrate more code onto ROCm as needed.

Co-authored-by: Ubuntu <ettao@ettao-amd-dev1.zvflicr54joexhdgnhvmxrxygg.phxx.internal.cloudapp.net>
2023-02-06 20:52:06 -08:00
..
github Add support for python 3.10 for onnxruntime-training cuda and cpu (#14100) 2023-02-02 11:32:41 -08:00
__init__.py
amd_hipify.py [ROCm] Enable Sampling Op UT on AMD (#14581) 2023-02-06 20:52:06 -08:00
build.py link mpi when either use_mpi or use_nccl enabled (#14467) 2023-02-03 20:11:50 +08:00
clean_docker_image_cache.py
coverage.py
gen_def.py [oneDNN] Improved thread handling (#13618) 2023-01-31 14:37:13 -08:00
get_docker_image.py Improve dependency management (#13523) 2022-12-01 09:51:59 -08:00
logger.py
op_registration_utils.py
op_registration_validator.py Update CUDA ArgMin/ArgMax op kernels to have end version 11 since opset 12+ is not supported yet. (#13983) 2022-12-21 19:01:00 -05:00
policheck_exclusions.xml Exculde hipify option from policheck (#13431) 2022-10-25 16:35:16 +08:00
reduce_op_kernels.py
replace_urls_in_deps.py Move C/C++ deps' URLs to deps.txt (#13769) 2022-11-29 18:06:35 -08:00
requirements.txt upgrade protobuf to 3.20.2 and onnx to 1.13 (#14279) 2023-01-31 12:55:09 -08:00
update_tsaoptions.py
upload_python_package_to_azure_storage.py