onnxruntime/orttraining
Vincent Wang c47f446f25
Support BFloat16 for Triton Codegen (#20353)
Previous implementation used numpy array and numpy data_type to store
constant value and data type, which is not support BFloat16 natively.
This PR is to switch to use torch tensor which supports BFloat16.
2024-04-18 17:15:11 +08:00
..
orttraining Support BFloat16 for Triton Codegen (#20353) 2024-04-18 17:15:11 +08:00
tools Bump ruff to 0.3.2 and black to 24 (#19878) 2024-03-13 10:00:32 -07:00