onnxruntime/orttraining/orttraining/python
Frank Dong 227c4419fc
add bf16 support for few ops (#20385)
### Description
Add bf16 support for below ops:
ConstantOfShape
Exp
Erf
convolution
PythonOp



### Motivation and Context
phimm model works on bf16, ORT need support bf16 on previous ops to work
with phimm on bf16
2024-04-25 11:28:34 -07:00
..
training add bf16 support for few ops (#20385) 2024-04-25 11:28:34 -07:00
orttraining_pybind_common.h Re-work global objects dependancies in pybind layer. (#14941) 2023-03-10 13:55:31 -08:00
orttraining_pybind_state.cc Check padding density by input of embedding module (#19821) 2024-04-10 18:45:51 +08:00
orttraining_python_module.cc [CUDA][ROCm][Training] Fix cuda/rocm provider info hash (#19398) 2024-02-05 14:35:57 -08:00
orttraining_python_module_eager.h Run clang-format in CI (#15524) 2023-04-18 09:26:58 -07:00
pt_patch.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00