saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-19 19:00:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/tools/ci_build
History
mindest a71dab691d
Implement BatchNormInternal for cuda (#8172) 
* correct batchnorm replacement output order;

remove bn replacement in grad graph builder

* update op defs and kernel class

* implement batch norm internal and grad.

* change saved_var into saved_inv_std

* cuda test case: bn internal

* remove redundant include

* fix comment; add support and UT for 1d input.

* exclude batch_norm_internal in amd_hipify

* run BNInternal UT for CUDA only

* fix CI error

* fix comment errors

* fix error

* add comment for inconsistency with cudnnBN doc

* additional comments for cudnnBN inconsistency
2021-07-28 16:04:49 +08:00
..
github [iOS Packaging] Update build definition (#8503) 2021-07-27 08:16:02 -07:00
wasm [wasm] emsdk: allow to install emscripten only (#7961) 2021-06-07 09:45:02 -07:00
__init__.py
amd_hipify.py Implement BatchNormInternal for cuda (#8172) 2021-07-28 16:04:49 +08:00
build.py Implement SparseTensor and infrastructure suppport and advance ONNX commit (#8038) 2021-07-22 15:24:36 -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 Move one function from cuda_provider_factory.h (#8407) 2021-07-19 17:55:59 -07:00
get_docker_image.py Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
logger.py
op_registration_utils.py Update Android AAR packaging pipeline script (#7559) 2021-05-04 11:13:33 -07: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
requirements.txt Add post-install command to build PyTorch CPP extensions from within onnxruntime package (#8027) 2021-06-28 18:11:58 -07:00
upload_python_package_to_azure_storage.py to support multiple cuda versions in published onnxruntime-training package (#7468) 2021-04-27 17:15:33 -07:00