Commits

Branch selector

 master

User selector

GleasonK

All time

Commit History

Commits on Jul 1, 2024

• [PJRT] Remove StableHLO bytecode patches now that VHLO is used.

  

  GleasonK

  authored and

  tensorflower-gardener

  committedJul 1, 2024
  de26c97

Commits on Jun 20, 2024

• [mhlo] Remove UnaryEinsumOp from MHLO

  

  GleasonK

  authored and

  tensorflower-gardener

  committedJun 20, 2024
  79ead0e

Commits on Jun 18, 2024

• Remove use of deprecated op UnaryEinsum.

  

  GleasonK

  authored and

  tensorflower-gardener

  committedJun 18, 2024
  3eb2492

• Adopt ConvertMlirHloToHloModule instead of passing in proto in PJRT

  

  GleasonK

  authored and

  tensorflower-gardener

  committedJun 18, 2024
  8f9cee4

Commits on Jun 15, 2024

• Add tests for mlir_to_hlo xla::Serialize method. Fix bug for programs emitting CHLO.

  

  GleasonK

  authored and

  tensorflower-gardener

  committedJun 15, 2024
  a2230df

Commits on Jun 14, 2024

• Integrate StableHLO at openxla/stablehlo@dd48ec58

  

  GleasonK

  authored and

  tensorflower-gardener

  committedJun 14, 2024
  46ee549

Commits on Jun 13, 2024

• [PJRT] Use SerializeUsingVersionedStablehlo for PJRT API v47+

  

  GleasonK

  authored and

  tensorflower-gardener

  committedJun 13, 2024
  d184a76

Commits on Jun 10, 2024

• Preserve HloModuleConfig in HLO<->MHLO.

  

  GleasonK

  authored and

  tensorflower-gardener

  committedJun 10, 2024
  6312122

• [HLO] Use .hlo for HLO text files, remove uses of .hlotxt.

  

  GleasonK

  authored and

  tensorflower-gardener

  committedJun 10, 2024
  20d558b

Commits on Jun 6, 2024

• [PJRT] Revert VHLO changes until after release

  

  GleasonK

  authored and

  tensorflower-gardener

  committedJun 6, 2024
  28608be

Commits on Jun 4, 2024

• [PJRT] Use SerializeUsingVersionedStablehlo for PJRT API version 41+

  

  GleasonK

  authored and

  tensorflower-gardener

  committedJun 4, 2024
  738fd1e

Commits on May 15, 2024

• Bump ODML to target StableHLO v1.0.0

  

  GleasonK

  authored and

  tensorflower-gardener

  committedMay 15, 2024
  59ad508

Commits on May 14, 2024

• Integrate StableHLO at openxla/stablehlo@797bee21

  

  GleasonK

  authored and

  tensorflower-gardener

  committedMay 14, 2024
  800071f

Commits on May 8, 2024

• [HLO] Allow zero-operand kAfterAll

  

  GleasonK

  authored and

  tensorflower-gardener

  committedMay 8, 2024
  a4f21e5

Commits on May 3, 2024

• Filter custom_call_schedule when encoding MHLO custom call in StablHLO

  

  GleasonK

  authored and

  tensorflower-gardener

  committedMay 3, 2024
  0790d1f

Commits on Apr 18, 2024

• Migrate StableHLO to use Properties

  

  GleasonK

  authored and

  tensorflower-gardener

  committedApr 18, 2024
  5420c31

Commits on Apr 15, 2024

• Integrate StableHLO at openxla/stablehlo@714d9aca

  

  GleasonK

  authored and

  tensorflower-gardener

  committedApr 15, 2024
  1e526f3

Commits on Apr 5, 2024

• Pass MLIR bytecode across XLA Extension boundary for JAX when converting StableHLO<->MHLO

  

  GleasonK

  authored and

  tensorflower-gardener

  committedApr 5, 2024
  f28921d

Commits on Mar 29, 2024

• [odml] Remove MHLO from CHLO->StableHLO lowering. Migrate JAX random lowering to StableHLO

  

  GleasonK

  authored and

  tensorflower-gardener

  committedMar 29, 2024
  b36854a

Commits on Mar 26, 2024

• Add CompositeOp to CustomOp conversion

  

  GleasonK

  authored and

  tensorflower-gardener

  committedMar 26, 2024
  d56e01e

Commits on Mar 21, 2024

• Integrate CHLO->StableHLO lowerings in XLA

  

  GleasonK

  authored and

  tensorflower-gardener

  committedMar 21, 2024
  76e23e8

• Fix JAX PjRT Clients using XLA pins from 12/2023.

  

  GleasonK

  authored and

  tensorflower-gardener

  committedMar 21, 2024
  3a5a9ec

Commits on Mar 19, 2024

• Only mark ops converted by pattern at illegal

  

  GleasonK

  authored and

  tensorflower-gardener

  committedMar 19, 2024
  7c6c923

Commits on Mar 12, 2024

• Integrate StableHLO at openxla/stablehlo@d978408a

  

  GleasonK

  authored and

  tensorflower-gardener

  committedMar 12, 2024
  0933918

Commits on Mar 7, 2024

• Don't legalize back to VHLO FuncOp during ODML->StableHLO

  

  GleasonK

  authored and

  tensorflower-gardener

  committedMar 7, 2024
  3509c48

Commits on Feb 29, 2024

• [odml] Change StableHLO->VHLO pass to use partial conversion instead of greedy rewriter to avoid CSE of arith constants.

  

  GleasonK

  authored and

  tensorflower-gardener

  committedFeb 29, 2024
  b4f292d

Commits on Feb 26, 2024

• Port Shape->StableHLO from GH

  

  GleasonK

  authored and

  tensorflower-gardener

  committedFeb 26, 2024
  22a418f

Commits on Feb 21, 2024

• Share StableHLO/MHLO pretty printers for ReduceOp and WhileOp

  

  GleasonK

  authored and

  tensorflower-gardener

  committedFeb 21, 2024
  658fa2f

• Allow VHLO to interoperate with TFLite dialect.

  

  GleasonK

  authored and

  tensorflower-gardener

  committedFeb 21, 2024
  c9c72c5

Commits on Feb 16, 2024

• Factor out MHLO custom_call op importer to ease longer term maintenance / serve as a more clear documentation of supported ops.

  

  GleasonK

  authored and

  tensorflower-gardener

  committedFeb 16, 2024
  d5dd9db

• Refactor binary broadcasting logic to make unbounded implicit broadcasting easier to implement.

  

  GleasonK

  authored and

  tensorflower-gardener

  committedFeb 16, 2024
  b0b274e

Commits on Feb 8, 2024

• Add support for real_dynamic_slice to HLO->MHLO converter

  

  GleasonK

  authored and

  tensorflower-gardener

  committedFeb 8, 2024
  207d36f

Commits on Feb 7, 2024

• Remove special TopK handling in SPMD partitioner

  

  GleasonK

  authored and

  tensorflower-gardener

  committedFeb 7, 2024
  c81766f

Commits on Feb 5, 2024

• Add MHLO handling to StablehloToTF ODML pipeline

  

  GleasonK

  authored and

  tensorflower-gardener

  committedFeb 5, 2024
  36c8dad

• Flip the switch and preserve TopK through XLA compilation

  

  GleasonK

  authored and

  tensorflower-gardener

  committedFeb 5, 2024
  bd9f92b

Pagination

PreviousNext