Public
  1. Public

onnxruntime

Public

Commits

Vincent WangGitHub
da86f6f4083
Vincent Wang authored and GitHub committed c47f446f25c
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.