Commits

Branch selector

 develop-upstream

User selector

tyb0807

Datepicker

All time

Commit History

Commits on Mar 31, 2024

• [xla:gpu] DUS support for generic custom call emission in DynamicAddressComputationFusion emitter

  Show description for 0cc13d2
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  0cc13d2

  Copy full SHA for 0cc13d2

• [xla:gpu] Generic custom call emission for DynamicAddressComputationFusion emitter

  Show description for e8cd9fd
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  e8cd9fd

  Copy full SHA for e8cd9fd

• [xla:gpu][NFC] Use the same helpers to get slices for GEMM and generic custom call emissions

  Show description for e6bb783
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  e6bb783

  Copy full SHA for e6bb783

Commits on Mar 30, 2024

• [xla:gpu][NFC] Explicitly rewrite AddressComputationFusion in custom call tests

  Show description for b2e993e
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  b2e993e

  Copy full SHA for b2e993e

• [xla:gpu][NFC] Make lambdas static functions for better reusability

  Show description for 92b03bd
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  92b03bd

  Copy full SHA for 92b03bd

• [xla:gpu] Unify static and dynamic slice cases for AddressComputationFusionRewriter

  Show description for fd091d4
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  fd091d4

  Copy full SHA for fd091d4

• [xla:gpu][NFC] Remove unused constexprs

  Show description for 7e9d002
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  7e9d002

  Copy full SHA for 7e9d002

• [xla:gpu] Unify GEMM emission for (Dynamic)AddressComputationFusion emitters

  Show description for ce7a132
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  ce7a132

  Copy full SHA for ce7a132

• [xla:gpu] AddressComputationFusionRewriter should run before other fusions

  Show description for 27bd888
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  27bd888

  Copy full SHA for 27bd888

Commits on Mar 29, 2024

• [xla:gpu] Add a version of HloPredicateIsOp for HloInstructionAdaptor

  Show description for 7810e73
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  7810e73

  Copy full SHA for 7810e73

• [xla:gpu][NFC] Simplify collect_slice_info

  Show description for c9cffcd
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  c9cffcd

  Copy full SHA for c9cffcd

• [xla:gpu][NFC] Use meaningful constexpr

  Show description for 06f2799
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  06f2799

  Copy full SHA for 06f2799

• [xla:gpu][NFC] Add AddressComputationThunk test with GEMM operands sharing the same buffer

  Show description for f2a78c0
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  f2a78c0

  Copy full SHA for f2a78c0

• [xla:gpu] Create fake buffer allocations for embedded thunk

  Show description for a829ac0
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  a829ac0

  Copy full SHA for a829ac0

• [xla:gpu][NFC] Use absl::Span more consistenly

  Show description for 903f741
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  903f741

  Copy full SHA for 903f741

Commits on Mar 28, 2024

• [xla:gpu][NFC] Refactor and rename in address_computation_fusion_rewriter for clarity

  Show description for 0722946
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  0722946

  Copy full SHA for 0722946

• [xla:gpu] Address computation thunk should not be inside command buffer

  Show description for 48bb29f
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  48bb29f

  Copy full SHA for 48bb29f

• [xla:gpu] Sort sliced operand paths on the fly

  Show description for 55cae5b
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  55cae5b

  Copy full SHA for 55cae5b

• [xla:gpu][NFC] Refactor address_computation_fusion_rewriter

  Show description for 5f92c14
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  5f92c14

  Copy full SHA for 5f92c14

• [xla:gpu][NFC] Add test for nested tuple custom call

  Show description for 8831b76
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  8831b76

  Copy full SHA for 8831b76

Commits on Mar 27, 2024

• [xla:gpu] Support tupled GEMM with DUS

  Show description for 1919e6c
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  1919e6c

  Copy full SHA for 1919e6c

Commits on Mar 25, 2024

• [xla:gpu] Handle OOB start indices in AddressComputationThunk

  Show description for d04a630
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  d04a630

  Copy full SHA for d04a630

• [xla:gpu][NFC] Add update accessor to HloDynamicUpdateSliceInstruction API

  Show description for e8086c7
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  e8086c7

  Copy full SHA for e8086c7

• [xla:gpu] Support DUS for AddressComputationFusionRewriter

  Show description for c76b77e
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  c76b77e

  Copy full SHA for c76b77e

• [xla:gpu] Support dynamic offset size in AddressComputationThunk

  Show description for 1b41fda
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  1b41fda

  Copy full SHA for 1b41fda

• [xla:gpu][NFC] Rename for consistency

  Show description for acc308a
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  acc308a

  Copy full SHA for acc308a

• [xla:gpu] Use fusion shape instead of hero shape when emit AddressComputationThunk

  Show description for 96091ee
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  96091ee

  Copy full SHA for 96091ee

Commits on Mar 22, 2024

• [xla:gpu] Support dynamic slices for AddressComputationFusionRewriter

  Show description for 702568d
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  702568d

  Copy full SHA for 702568d

• [xla:gpu] Factor out AddressComputationFusionRewriter::Run to handle dynamic slices

  Show description for b27ce64
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  b27ce64

  Copy full SHA for b27ce64

• [XLA:GPU][NFC] Remove unused variable

  Show description for 9d1e995
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  9d1e995

  Copy full SHA for 9d1e995

• [XLA:GPU][NFC] Fix style in in_place_dynamic_update_slice_mlir.cc for consistency

  Show description for 5abfc19
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  5abfc19

  Copy full SHA for 5abfc19

• [xla:gpu][NFC] Documenting address_computation_fusion_rewriter

  Show description for 168ee8b
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  168ee8b

  Copy full SHA for 168ee8b

• [XLA:GPU] Add MLIR-based DUS emitter.

  Show description for baebf00
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  baebf00

  Copy full SHA for baebf00

• [xla:gpu][NFC] Simplify AddressComputationFusionRewriter::Run

  Show description for 2a1f794
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  2a1f794

  Copy full SHA for 2a1f794

• [XLA:GPU][NFC] Remove unnecessary check in MlirConcatenateFusion emitter

  Show description for 1666096
  

  tyb0807

  authored and

  tensorflower-gardener

  committed
  1666096

  Copy full SHA for 1666096

Pagination

Previous

Next