onnxruntime
Using a map of of ops to stages as input of partition function.
#5940
Merged
Overview
Commits
32
Changes
View On GitHub

Using a map of of ops to stages as input of partition function. #5940

wschin merged 32 commits into microsoft:master from jupvfranco:jufranc/stage-map-partition
jupvfranco
New partition algorithm running before AD
9ba761e5
Convert cut_group_info into device map. Work in progress -- works for…
cf2e9c85
Removing code for partition of bwd graphs
23e583c8
Remove old code
580bb883
Adding some verification code
6d0ce129
Handle Shared Initializer
8531b4fc
Renaming rank with stage
119d8fe6
Added first unit test
9a2624a6
new test
3c4ba94c
redundant check
a2fe9aa7
undo change in bert
40f8dba1
Moved cut-based partition to testing utils file
c6fb5c8c
tlh20
tlh20
azure-pipelines
azure-pipelines
New conversion function and tests
111c1dfd
minor
63f3f00e
remove test that is not needed2
e887fe34
xzhu1900
azure-pipelines
xzhu1900
azure-pipelines
alberto-magni
alberto-magni commented on 2020-12-02
alberto-magni
alberto-magni commented on 2020-12-02
alberto-magni
alberto-magni commented on 2020-12-02
jupvfranco jupvfranco changed the title [WiP] Using a map of of ops to stages as input of partition function. Using a map of of ops to stages as input of partition function. 5 years ago
improve GetDeviceAssignment and PR comments
61c0beb8
awf
awf commented on 2020-12-03
minor changes
455472b2
PR comments
a60a82ba
jupvfranco jupvfranco marked this pull request as ready for review 5 years ago
jupvfranco jupvfranco requested a review 5 years ago
wschin
wschin commented on 2020-12-03
wschin
wschin commented on 2020-12-03
wschin
wschin commented on 2020-12-03
wschin
wschin commented on 2020-12-03
wschin
wschin commented on 2020-12-03
wschin
wschin commented on 2020-12-03
wschin
wschin commented on 2020-12-03
wschin
wschin commented on 2020-12-03
wschin
wschin commented on 2020-12-03
wschin
wschin commented on 2020-12-03
wschin
wschin commented on 2020-12-03
wschin
wschin commented on 2020-12-03
wschin
wschin commented on 2020-12-03
improving documentation and variable naming
516e52ff
jupvfranco
azure-pipelines
jupvfranco
azure-pipelines
add documentation
14d86eac
Variable naming and docs
f4291d40
more doc improvements
f327239f
more doc improvements
de15ac84
jupvfranco
azure-pipelines
jupvfranco
azure-pipelines
missing static cast
0da18caf
Fix test file for windows
166c0788
alberto-magni
alberto-magni commented on 2020-12-10
Fix test file for windows
0bcaadc1
Fix test file for windows
fc431c38
stage id is not the same as rank id
66c7486e
wschin
wschin commented on 2020-12-15
wschin
wschin commented on 2020-12-15
wschin
wschin commented on 2020-12-15
wschin
wschin commented on 2020-12-15
wschin
wschin commented on 2020-12-15
wschin
wschin commented on 2020-12-16
wschin
wschin commented on 2020-12-16
wschin
wschin commented on 2020-12-16
wschin
wschin commented on 2020-12-16
wschin
wschin commented on 2020-12-16
wschin
wschin commented on 2020-12-16
wschin
wschin commented on 2020-12-16
wschin
wschin commented on 2020-12-16
wschin
wschin commented on 2020-12-16
wschin
wschin commented on 2020-12-16
wschin
wschin commented on 2020-12-16
wschin
wschin commented on 2020-12-16
wschin
wschin commented on 2020-12-16
wschin
wschin commented on 2020-12-16
wschin
wschin commented on 2020-12-16
wschin
wschin commented on 2020-12-16
PR comments
a40c0429
PR comments
6fa0cd88
More comments
b4da5f59
More comments
8fbfc48a
wschin
wschin approved these changes on 2020-12-17
wschin wschin merged 36c03b32 into master 5 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum