davinwang's picture
compatible with DirectML/ROCm
5bc5aff