Commits

Branch selector

 master

User selector

Tongfei-Guo

All time

Commit History

Commits on Jul 1, 2024

• [XLA:SPMD] Do broadcast splitting before sharding propagation.

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedJul 1, 2024
  9467e3a

Commits on Jun 27, 2024

• [XLA:SPMD] Propagate conditional's parameter shardings back to conditional's operands to avoid discrepancy of input/parameter shardings.

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedJun 27, 2024
  87c1321

Commits on May 18, 2024

• [XLA:SPMD] Support partitioning kCall.

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedMay 18, 2024
  b409682

Commits on May 16, 2024

• Allow overriding some methods in all_gather_decomposer.

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedMay 16, 2024
  ebff15d

Commits on May 10, 2024

• Allow setting operand_shapes_with_layout on custom call instruction.

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedMay 10, 2024
  389f3d5

Commits on May 8, 2024

• [XLA:SPMD] Use the default way to call once and shard barrier custom-call registration instead of always linking it.

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedMay 8, 2024
  69b50ff

Commits on May 3, 2024

• [XLA:SPMD] Do not run shard group propagation if it's CSE prevention.

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedMay 3, 2024
  f7ab159

Commits on May 2, 2024

• [XLA] Add an option to dump large constants.

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedMay 2, 2024
  fb8cd15

Commits on May 1, 2024

• [XLA:SPMD] Support shard-as propagation with unspecified_dims.

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedMay 1, 2024
  6272368

• [XLA:SPMD] support sharding barrier (1/N).

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedMay 1, 2024
  eaa0b19

• [XLA:SPMD] support sharding barrier (0/N).

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedMay 1, 2024
  0cf7880

Commits on Apr 15, 2024

• [XLA:SPMD] Add DUS to aggressiveness 0.

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedApr 15, 2024
  e246f06

Commits on Apr 12, 2024

• [XLA:AlgebraicSimplifier] Bugfix on gather simplification.

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedApr 12, 2024
  7ecd526

Commits on Apr 11, 2024

• [XLA:AlgebraicSimplifier] Simplify the following gather patterns:

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedApr 11, 2024
  8acab76

Commits on Mar 14, 2024

• [XLA:SPMD] Fix bug in variadic scatter where shared operands are group-partitioned more than once.

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedMar 14, 2024
  8cc0e31

Commits on Jan 10, 2024

• [XLA:SPMD] Restructure dot partitioning into individual methods.(NFC) (2/N)

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedJan 10, 2024
  6229b45

Commits on Jan 4, 2024

• [XLA:SPMD] Allow fast shallow copy of SPMDPartitioningVisitor to reduce the overhead of compilation time from creating a new fake visitor every time.

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedJan 4, 2024
  a80fccc

• [XLA:SPMD] Some improvements in gather/scatter.

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedJan 4, 2024
  d5d6777

Commits on Jan 3, 2024

• [XLA:SPMD] Support evaluation of memory and communication cost for any arbitrary partitioning methods by dry-run the partitioning on that op only, and calculate the costs. (1/N)

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedJan 3, 2024
  bbd8fd7

Commits on Dec 20, 2023

• Allow shard like on different data types and tilings as long as shapes agree.

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedDec 20, 2023
  7c6f324

• Fix bug in shard_like where exclude parameters from shard_group if allow_spmd_sharding_propagation_to_parameters is false.

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedDec 20, 2023
  af61e8f

Commits on Dec 16, 2023

• Allow shard_like on input/output with custom-call sharding annotation.

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedDec 16, 2023
  17cc52c

Commits on Nov 3, 2023

• [XLA:Collective] Fix bug in reduce-scatter matching where AR->Reshape->DUS is not correctly recognized.

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedNov 3, 2023
  65bf548

Commits on Oct 12, 2023

• Fix test breaking

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedOct 12, 2023
  18f9ee9

Commits on Oct 11, 2023

• [XLA:SPMD] Improve tupled shape manual sharding propagation.

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedOct 11, 2023
  790d784

Commits on Oct 5, 2023

• [XLA:SPMD] Fix wrong broadcast shape in the Gather trivial sliced dimension partitioning.

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedOct 5, 2023
  8541a32

Commits on Oct 3, 2023

• [XLA:SPMD] Supports splitting Gather/Scatter sharding dimensions so that a shard dimension can be used for more than 1 partitioning methods. This allows us to reduce communication such that we don'…

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedOct 3, 2023
  0053091

Commits on Oct 2, 2023

• [XLA:SPMD] Support sharding propagation/partitioning for variadic Scatter.

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedOct 2, 2023
  ba46683

• [XLA:SPMD] Bug fix that empty tuple sharding shouldn't be a shard group.

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedOct 2, 2023
  2a64b46

Commits on Sep 29, 2023

• Add gather test coverage for some bug.

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedSep 29, 2023
  cee7495

Commits on Sep 11, 2023

• [SPMD] Add sharding propagation logics/tests for shard_like/as. (3/N)

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedSep 11, 2023
  af73df5

• [SPMD] Create a placeholder sharding of type UNKNOWN with the lowest precedence that can be overwriten by any other shardings. (2/N)

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedSep 11, 2023
  ebdbaa3

Commits on Aug 31, 2023

• [SPMD] Prevent DCE on sharding custom-call to allow a dangling sharding custom-call attached to an op.(1/N)

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedAug 31, 2023
  429b183

• [SPMD] Support shard like interface in XLA and JAX (0/N).

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedAug 31, 2023
  ae366f7

Commits on Aug 15, 2023

• [XLA] avoid brute force search device assignment repeated when flattening replica groups.

  

  Tongfei-Guo

  authored and

  tensorflower-gardener

  committedAug 15, 2023
  bcfa044

Pagination

PreviousNext