onnxruntime/orttraining
Vincent Wang c07a3b869c
Triton Codegen for ORTModule (#15831)
Fuse connected elementwise and reduce Ops to TritonOp and codegen triton
code to run the kernel.

This PR is co-edited by @wejoncy and @er3x3
2023-07-13 18:17:58 +08:00
..
orttraining Triton Codegen for ORTModule (#15831) 2023-07-13 18:17:58 +08:00
pytorch_frontend_examples Enable pylint and numpy rules (#15218) 2023-03-27 20:37:53 -07:00
tools [ROCm] Move ROCm build step on CPU only machine (#16596) 2023-07-10 11:55:10 +08:00