Commits

Branch selector

 master

User selector

farzinhoushmand

All time

Commit History

Commits on Jun 28, 2024

• Add more custom pattern to hlo_unstacker pass.

  

  farzinhoushmand

  authored and

  tensorflower-gardener

  committedJun 28, 2024
  5cf5bd1

Commits on Jun 22, 2024

• Add GetUniqueGteInstruction to hlo_query utility file.

  

  farzinhoushmand

  authored and

  tensorflower-gardener

  committedJun 22, 2024
  069e051

Commits on Jun 5, 2024

• Refactor loop unroller pass.

  

  farzinhoushmand

  authored and

  tensorflower-gardener

  committedJun 5, 2024
  955d406

Commits on May 30, 2024

• First cl to implement a pass to unstack loop operands.

  

  farzinhoushmand

  authored and

  tensorflower-gardener

  committedMay 30, 2024
  f39297b

• Change MatchShapeCoveringDynamicIndexInstruction to return the dynamic index if matched.

  

  farzinhoushmand

  authored and

  tensorflower-gardener

  committedMay 30, 2024
  e9a3e05

Commits on May 29, 2024

• Change IsCollectiveWithChannelId to return the instruction with channel_id or nullptr.

  

  farzinhoushmand

  authored and

  tensorflower-gardener

  committedMay 29, 2024
  1d40455

Commits on May 22, 2024

• Some refactoring to while_loop_unroller.cc

  

  farzinhoushmand

  authored and

  tensorflower-gardener

  committedMay 22, 2024
  eeeeda0

Commits on May 21, 2024

• Move MatchShapeCoveringDynamicIndex function to while_loop_unroller.h and update other usages to use the same method.

  

  farzinhoushmand

  authored and

  tensorflower-gardener

  committedMay 21, 2024
  bb20154

• Introduce MakeConstantWithShape in hlo_creation_util file.

  

  farzinhoushmand

  authored and

  tensorflower-gardener

  committedMay 21, 2024
  97a794b

Commits on May 10, 2024

• Make unroller utility functions static.

  

  farzinhoushmand

  authored and

  tensorflower-gardener

  committedMay 10, 2024
  1689e67

Commits on May 4, 2024

• Add a pass to unify the accumulator and the input of the jax.scan loops.

  

  farzinhoushmand

  authored and

  tensorflower-gardener

  committedMay 4, 2024
  583f81f

• Some refactoring to WhileLoopUnroller pass.

  

  farzinhoushmand

  authored and

  tensorflower-gardener

  committedMay 4, 2024
  3a0419d

Commits on Apr 23, 2024

• Call IsCollective when looking for collectives with channel id.

  

  farzinhoushmand

  authored and

  tensorflower-gardener

  committedApr 23, 2024
  a60b0b6