Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Nov 10, 2023

• Remove erroneous TSL change

  jakeharmon8

  committed
  83ccf2a

  Copy full SHA for 83ccf2a

• Google internal CI workaround

  Show description for d6db38a
  jakeharmon8

  committed
  d6db38a

  Copy full SHA for d6db38a

• Testing internal Google CI - please ignore

  jakeharmon8

  committed
  7f322ef

  Copy full SHA for 7f322ef

Commits on Nov 9, 2023

• Roundtrip unbounded dimension sizes between MHLO and HLO

  Show description for 24c4e6b
  

  ghpvnist

  authored and

  copybara-github

  committed
  24c4e6b

  Copy full SHA for 24c4e6b

• Reverts 633ed04f692ee349247d69f3aedfd8884217f9f0

  Show description for dc0c102
  

  hyeontaek

  authored and

  copybara-github

  committed
  dc0c102

  Copy full SHA for dc0c102

• [PJRT C API] Plumb PJRT C API version to JAX python client.

  Show description for 972846d
  

  Jieying Luo

  authored and

  copybara-github

  committed
  972846d

  Copy full SHA for 972846d

• Add comment to enable_tensor_float_32_execution

  Show description for ab3562d
  

  tdanyluk

  authored and

  copybara-github

  committed
  ab3562d

  Copy full SHA for ab3562d

• [XLA:GPU] Ignore tilings which are uncompilable due to RegisterAllocation failure

  Show description for fb6548c
  

  tdanyluk

  authored and

  copybara-github

  committed
  fb6548c

  Copy full SHA for fb6548c

• Integrate StableHLO at openxla/stablehlo@897ba37f

  Show description for 1781adb
  

  Eugene Burmako

  authored and

  copybara-github

  committed
  1781adb

  Copy full SHA for 1781adb

• Update TFRT dependency to use revision

  Show description for 2870303
  

  tensorflower-gardener

  authored and

  copybara-github

  committed
  2870303

  Copy full SHA for 2870303

• Small cleanup of tf_op_utils.

  Show description for 0bc02d7
  

  tensorflower-gardener

  authored and

  copybara-github

  committed
  0bc02d7

  Copy full SHA for 0bc02d7

• Integrate LLVM at llvm/llvm-project@87b00ef35a00

  Show description for 46134b9
  

  durin42

  authored and

  copybara-github

  committed
  46134b9

  Copy full SHA for 46134b9

• Reverts fa114efd9701ee6876834401c62c7baaf58764df

  Show description for 4a06c21
  

  ddunl

  authored and

  copybara-github

  committed
  4a06c21

  Copy full SHA for 4a06c21

• [XLA] Expose method for parsing Layout strings

  Show description for 84ebd1b
  

  skye

  authored and

  copybara-github

  committed
  84ebd1b

  Copy full SHA for 84ebd1b

• [XLA:GPU] Check tensor_float_32_execution_enabled() in Triton codegen too

  Show description for 771e381
  

  tdanyluk

  authored and

  copybara-github

  committed
  771e381

  Copy full SHA for 771e381

• [xla_compile] Support outputting compilation metadata in a structured manner.

  Show description for 5705821
  

  pizzud

  authored and

  copybara-github

  committed
  5705821

  Copy full SHA for 5705821

• [Tile Analysis] Add a utility to compose tiles.

  Show description for cdf659f
  

  pifon2a

  authored and

  copybara-github

  committed
  cdf659f

  Copy full SHA for cdf659f

• [triton] Change all pointers to opaque.

  Show description for bd30aaf
  

  olegshyshkov

  authored and

  copybara-github

  committed
  bd30aaf

  Copy full SHA for bd30aaf

• Update visibility for gomlx.

  Show description for 5477dc3
  

  tensorflower-gardener

  authored and

  copybara-github

  committed
  5477dc3

  Copy full SHA for 5477dc3

• [xla:gpu] Correctly get allocation slice for fusion roots in kReduction fusions #6528

  Show description for a22bcfd
  

  anlunx

  authored and

  copybara-github

  committed
  a22bcfd

  Copy full SHA for a22bcfd

• [JAX] Release GIL while disassembling single-device IFRT arrays

  Show description for 633ed04
  

  hyeontaek

  authored and

  copybara-github

  committed
  633ed04

  Copy full SHA for 633ed04

• [xla:gpu] Use CUDA C++ to define MakeBatchPointers kernel #6834

  Show description for 74cc001
  

  ezhulenev

  authored and

  copybara-github

  committed
  74cc001

  Copy full SHA for 74cc001

• [stream_executor] Add support for user-defined kernel arguments packing function to se::Kernel

  Show description for 6a0016f
  

  ezhulenev

  authored and

  copybara-github

  committed
  6a0016f

  Copy full SHA for 6a0016f

• Use per PJRT device lock for tracking and untracking buffer in TfPjRtClient.

  Show description for 2afe1e4
  

  Jieying Luo

  authored and

  copybara-github

  committed
  2afe1e4

  Copy full SHA for 2afe1e4

• [jax] Better distinguish between `pathways` and `pathways_ifrt` in xla_client_test.py

  Show description for f3aae35
  

  skye

  authored and

  copybara-github

  committed
  f3aae35

  Copy full SHA for f3aae35

Commits on Nov 8, 2023

• [jax] Expose executable output layouts

  Show description for feb89d9
  

  skye

  authored and

  copybara-github

  committed
  feb89d9

  Copy full SHA for feb89d9

• [triton] Add elem type to SharedMemoryObject.

  Show description for 70d065f
  

  olegshyshkov

  authored and

  copybara-github

  committed
  70d065f

  Copy full SHA for 70d065f

• Introduce stablehlo/experimental

  Show description for d09d73c
  

  Eugene Burmako

  authored and

  copybara-github

  committed
  d09d73c

  Copy full SHA for d09d73c

• PR #6850: [ROCM] fixing rocm buildbrakes 23-11-08

  Show description for 5af20c8
  

  pemeliya

  authored and

  copybara-github

  committed
  5af20c8

  Copy full SHA for 5af20c8

• [stream_executor] Fix overload resolution for DeviceMemory<T>*

  Show description for 1bca718
  

  ezhulenev

  authored and

  copybara-github

  committed
  1bca718

  Copy full SHA for 1bca718

• [xla:gpu] Add missing import in buffer_comparator.cu.cc

  Show description for 5bdd481
  

  tensorflower-gardener

  authored and

  copybara-github

  committed
  5bdd481

  Copy full SHA for 5bdd481

• Move the loop body of the TopK TransformToCustomCall into its own function.

  Show description for e269bbf
  

  tensorflower-gardener

  authored and

  copybara-github

  committed
  e269bbf

  Copy full SHA for e269bbf

• Adds a heuristic to reduce the number of mesh shapes to try when try_multiple_mesh_shapes is true.

  Show description for 38ab685
  

  tensorflower-gardener

  authored and

  copybara-github

  committed
  38ab685

  Copy full SHA for 38ab685

• Fix JAX build script to not use `bazel_wrapper.py`, add extra options required for GPU build

  Show description for 4619cf6
  

  ddunl

  authored and

  copybara-github

  committed
  4619cf6

  Copy full SHA for 4619cf6

• Make sure `benchmark-test` tagged targets are excluded from all test configs.

  Show description for ef5cda7
  

  tensorflower-gardener

  authored and

  copybara-github

  committed
  ef5cda7

  Copy full SHA for ef5cda7

Pagination

Previous

Next