1. 08 Oct, 2020 1 commit
    • Min Xu's avatar
      [test] Add unittest for checkpoint & DDP (#126) · 6658be22
      Min Xu authored
      * Add unittest for checkpoint & DDP
      
      - this change adds test cases to reproduce the error with checkpoint & DDP
      - mandeep mentioned that there is also deadlock in this case, but this
        change doesn't cover that.
      - we cover cases where weight sharing is OK
      - however, same module multiple checkpoint or find_unused_parameters are
        both not OK
      
      * added norm checks
      6658be22
  2. 17 Sep, 2020 1 commit
    • Tom Birch's avatar
      Multi-process pipe (#90) · 63f7796a
      Tom Birch authored
      Adds support for distributing pipeline stages across multiple processes (and therefore multiple machines)
      * Adds a style argument to the Pipe constructor, defaulting to PipelineStyle.SingleProcess, but also supporting PipelineStyle.MultiProcess
      * Added support for lazy construction of modules (see lazy_construction for an example)
      * Added two implementations of inter-process communication: one based on rpc with globally visible queues, one based on send/recv
      * Copied all the relevant tests from tests/pipe to tests/pipe_process and modified them to exercise PipelineStyle.MultiProcess
      63f7796a
  3. 08 Jul, 2020 1 commit