onnxruntime
Add CPU allocation test for multiple GPU distributed run
#15829
Merged

Add CPU allocation test for multiple GPU distributed run #15829

pengwa merged 6 commits into main from pengwa/add_ut
pengwa
pengwa fix cpu device id to be 0
2f01fd2f
pengwa minor
506d20b1
pengwa lint
96e4a9bf
pengwa add ut for distributed tests using both CPU and CUDA for output alloc…
0d8e8709
pengwa Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
5b0322bb
pengwa format
7f81c245
pengwa pengwa requested a review from askhade askhade 2 years ago
pengwa pengwa requested a review from baijumeswani baijumeswani 2 years ago
pengwa pengwa requested a review from PeixuanZuo PeixuanZuo 2 years ago
pengwa pengwa requested a review from satyajandhyala satyajandhyala 2 years ago
pengwa pengwa added training
baijumeswani
baijumeswani approved these changes on 2023-05-08
pengwa
pengwa pengwa merged 003c7d3e into main 2 years ago
pengwa pengwa deleted the pengwa/add_ut branch 2 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone