onnxruntime
Using a map of of ops to stages as input of partition function.
#5940
Merged
Overview
Commits
32
Changes
View On GitHub
Commits
  • New partition algorithm running before AD
    Juliana Franco committed 5 years ago
  • Convert cut_group_info into device map. Work in progress -- works for bert-tiny with pp=2
    Juliana Franco committed 5 years ago
  • Removing code for partition of bwd graphs
    Juliana Franco committed 5 years ago
  • Remove old code
    Juliana Franco committed 5 years ago
  • Adding some verification code
    Juliana Franco committed 5 years ago
  • Handle Shared Initializer
    Juliana Franco committed 5 years ago
  • Renaming rank with stage
    Juliana Franco committed 5 years ago
  • Added first unit test
    Juliana Franco committed 5 years ago
  • new test
    Juliana Franco committed 5 years ago
  • redundant check
    Juliana Franco committed 5 years ago
  • undo change in bert
    Juliana Franco committed 5 years ago
  • Moved cut-based partition to testing utils file
    Juliana Franco committed 5 years ago
  • New conversion function and tests
    Juliana Franco committed 5 years ago
  • minor
    Juliana Franco committed 5 years ago
  • remove test that is not needed2
    Juliana Franco committed 5 years ago
  • improve GetDeviceAssignment and PR comments
    Juliana Franco committed 5 years ago
  • minor changes
    Juliana Franco committed 5 years ago
  • PR comments
    Juliana Franco committed 5 years ago
  • improving documentation and variable naming
    Juliana Franco committed 5 years ago
  • add documentation
    Juliana Franco committed 5 years ago
  • Variable naming and docs
    Juliana Franco committed 5 years ago
  • more doc improvements
    Juliana Franco committed 5 years ago
  • more doc improvements
    Juliana Franco committed 5 years ago
  • missing static cast
    Juliana Franco committed 5 years ago
  • Fix test file for windows
    Juliana Franco committed 5 years ago
  • Fix test file for windows
    Juliana Franco committed 5 years ago
  • Fix test file for windows
    Juliana Franco committed 5 years ago
  • stage id is not the same as rank id
    Juliana Franco committed 5 years ago
  • PR comments
    Juliana Franco committed 5 years ago
  • PR comments
    Juliana Franco committed 5 years ago
  • More comments
    Juliana Franco committed 5 years ago
  • More comments
    Juliana Franco committed 5 years ago
FAQ Terms Privacy Refunds Impressum
Loading