onnxruntime/orttraining
Vincent Wang d30c81d270
Add Symbolic Shape Hint to Triton Codegen Config (#20056)
Add symbolic shape hint to Triton codegen config so that we can avoid
unnecessary recompile when input shapes are keeping changing. Below
screenshot shows that with proper configuration, we can speed up the
training a lot by reducing unnecessary recompile.


![image](https://github.com/microsoft/onnxruntime/assets/11661208/699944d2-81cd-4c22-84e7-73a4fa0d2a28)
2024-03-25 15:05:02 +08:00
..
orttraining Add Symbolic Shape Hint to Triton Codegen Config (#20056) 2024-03-25 15:05:02 +08:00
tools Bump ruff to 0.3.2 and black to 24 (#19878) 2024-03-13 10:00:32 -07:00