onnxruntime

d271a010 - Chenta/partition stream with device (#12656)

Chenta/partition stream with device (#12656) * partition stream based on device * fix win warning * fix training build * fix training warning * fix training Co-authored-by: Cheng Tang <chenta@microsoft.com>