onnxruntime/tools/ci_build
Weixing Zhang a3f17c8b0d
update lamb and GatherGrad kernel for ROCm EP (#7184)
With ROCm4.1, the CUDA implementation of Lamb and GatherGrad can be
utilized for ROCm EP.
2021-04-02 09:02:49 -07:00
..
github Check kernel def hashes (#7120) 2021-04-01 17:42:58 -07:00
__init__.py Add validation of op registrations (#5817) 2020-11-17 10:44:09 -08:00
amd_hipify.py update lamb and GatherGrad kernel for ROCm EP (#7184) 2021-04-02 09:02:49 -07:00
build.py Add Android AAR packaging script for ORT-Mobile (#7138) 2021-03-30 18:42:18 -07:00
clean_docker_image_cache.py Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
coverage.py Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
gen_def.py
get_docker_image.py Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
logger.py Cache build docker images in container registry. (#5811) 2020-11-17 17:02:24 -08:00
op_registration_utils.py Support required types when excluding typed registrations (#6871) 2021-03-08 08:22:07 -08:00
op_registration_validator.py Support specifying globally allowed types from build script (#6677) 2021-02-22 14:05:00 -08:00
reduce_op_kernels.py Support required types when excluding typed registrations (#6871) 2021-03-08 08:22:07 -08:00