Code

Control flow support#124

Code

Open

HeydrichBeillschmidt wants to merge 13 commits intoalpa-projects:masteralpa-projects/tensorflow-alpa:masterfrom HeydrichBeillschmidt:control-flow-supportHeydrichBeillschmidt/tensorflow-alpa:control-flow-supportCopy head branch name to clipboard

+10,742-131

Commits

Commits on Aug 30, 2022

• [NCCL] Statically link nccl from cupy

  merrymercy

  committed
  86b81f6

  Copy full SHA for 86b81f6

• Export memory stats from tf allocator

  merrymercy

  committed
  57fbe1f

  Copy full SHA for 57fbe1f

• Add random number to temp filename to avoid name conflcitions

  merrymercy

  committed
  d6e60b8

  Copy full SHA for d6e60b8

• Add pass_context to pass python arguments to c++

  merrymercy

  committed
  7495d93

  Copy full SHA for 7495d93

• Add bypass_device_assignment_check and rng seed setter

  merrymercy

  committed
  693b063

  Copy full SHA for 693b063

• Add auto-sharding related passes

  Show description for b58c711
  

  3 peoplecommitted
  b58c711

  Copy full SHA for b58c711

• Add all changes to GPU backend

  Show description for b66aedf
  

  3 peoplecommitted
  b66aedf

  Copy full SHA for b66aedf

• Add all changes under service

  merrymercy

  committed
  b66746b

  Copy full SHA for b66746b

• Add all changes to the xla_compiler and client's python interface

  Show description for 9c6f230
  

  3 peoplecommitted
  9c6f230

  Copy full SHA for 9c6f230

• Use identity marker in optimization barrier expander

  merrymercy

  committed
  6e31c0f

  Copy full SHA for 6e31c0f

Commits on Sep 3, 2022

• Convert pipeline_marker and OptimizationBarrier to bitcast (alpa-projects#132)

  merrymercy

  authored
  0e5d1b0

  Copy full SHA for 0e5d1b0

Commits on Sep 8, 2022

• [WIP] rebasing based on master branch

  HeydrichBeillschmidt

  committed
  a904257

  Copy full SHA for a904257

Commits on Dec 14, 2022

• AdjusteDebugged kConditional

  HeydrichBeillschmidt

  committed
  d00f182

  Copy full SHA for d00f182