Commits

Branch selector

 main

User selector

skye

Datepicker

All time

Commit History

Commits on Nov 9, 2023

• [XLA] Expose method for parsing Layout strings

  Show description for 84ebd1b
  

  skye

  authored and

  copybara-github

  committed
  84ebd1b

  Copy full SHA for 84ebd1b

• [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

Commits on Nov 2, 2023

• PR #6710: [XLA:GPU] Refactor cub_sort_kernel.cu.cc to not use abseil

  Show description for 7050245
  

  skye

  authored and

  copybara-github

  committed
  7050245

  Copy full SHA for 7050245

Commits on Nov 1, 2023

• [PJRT:C] Add forwards compat version guard for PJRT_Client_TopologyDescription call

  Show description for 1a9a09f
  

  skye

  authored and

  copybara-github

  committed
  1a9a09f

  Copy full SHA for 1a9a09f

Commits on Oct 31, 2023

• [PJRT:C] Fix memory management of PJRT_TopologyDescriptions in client

  Show description for c3a0fce
  

  skye

  authored and

  copybara-github

  committed
  c3a0fce

  Copy full SHA for c3a0fce

Commits on Oct 27, 2023

• [PJRT:C] Add const to a bunch of lists and value types in the header file.

  Show description for b215a12
  

  skye

  authored and

  copybara-github

  committed
  b215a12

  Copy full SHA for b215a12

• [PJRT:C] Implement PjRtCApiClient::GetTopologyDescription

  Show description for 5c449c5
  

  skye

  authored and

  copybara-github

  committed
  5c449c5

  Copy full SHA for 5c449c5

• [PJRT:C] Add PJRT_Client_TopologyDescription

  Show description for 39f5daf
  

  skye

  authored and

  copybara-github

  committed
  39f5daf

  Copy full SHA for 39f5daf

Commits on Oct 24, 2023

• [PJRT:C] Move class declarations around in pjrt_c_api_client.h and pjrt_c_api_wrapper_impl.h

  Show description for a1ca82d
  

  skye

  authored and

  copybara-github

  committed
  a1ca82d

  Copy full SHA for a1ca82d

Commits on Oct 13, 2023

• [jax] Expose executable input layouts and add test for setting input layouts.

  Show description for bff73b1
  

  skye

  authored and

  copybara-github

  committed
  bff73b1

  Copy full SHA for bff73b1

Commits on Oct 3, 2023

• [jax] Add pathways_ifrt xla_client_test config

  Show description for 7ed4639
  

  skye

  authored and

  copybara-github

  committed
  7ed4639

  Copy full SHA for 7ed4639

Commits on Sep 1, 2023

• [jax] Remove skip_checks default argument when creating PyArray.

  Show description for f63d0fb
  

  skye

  authored and

  copybara-github

  committed
  f63d0fb

  Copy full SHA for f63d0fb

Commits on Aug 29, 2023

• [JAX] Implement importing external dlpack-aware Python arrays.

  Show description for f71f950
  

  skye

  authored and

  copybara-github

  committed
  f71f950

  Copy full SHA for f71f950

• [PJRT:C] Small follow-ups to PJRT_Buffer_Memory

  Show description for 501850d
  

  skye

  authored and

  copybara-github

  committed
  501850d

  Copy full SHA for 501850d

• [PJRT:C] Remove xla::Shape getter and related logic from PJRT C API

  Show description for e30a46d
  

  skye

  authored and

  copybara-github

  committed
  e30a46d

  Copy full SHA for e30a46d

Commits on Aug 28, 2023

• [PJRT] Add `stream` argument to PjRtClient::CreateViewOfDeviceBuffer

  Show description for 0cbde3b
  

  skye

  authored and

  copybara-github

  committed
  0cbde3b

  Copy full SHA for 0cbde3b

Commits on Aug 22, 2023

• Fix dlpack breakage introduced in https://github.com/openxla/xla/commit/0b3cbfe4bc7cd68dc20924bd878cdfb2faa1a169.

  Show description for f67276e
  

  skye

  authored and

  copybara-github

  committed
  f67276e

  Copy full SHA for f67276e

Commits on Aug 21, 2023

• [JAX] Add xla_client.Client.device_from_local_hardware_id method.

  Show description for 49133d2
  

  skye

  authored and

  copybara-github

  committed
  49133d2

  Copy full SHA for 49133d2

Commits on Aug 18, 2023

• [JAX] Implement the `stream` argument to jax.Array.__dlpack__ for CUDA GPU

  Show description for 0b3cbfe
  

  skye

  authored and

  copybara-github

  committed
  0b3cbfe

  Copy full SHA for 0b3cbfe

Commits on Aug 16, 2023

• Rollback of https://github.com/openxla/xla/commit/d8ffc750ad249f362fc193e317f1ea436dc2d7c6

  Show description for a9fbd81
  

  skye

  authored and

  copybara-github

  committed
  a9fbd81

  Copy full SHA for a9fbd81

Commits on Aug 15, 2023

• [jax] Expose Device.local_hardware_id

  Show description for 3ca8f94
  

  skye

  authored and

  copybara-github

  committed
  3ca8f94

  Copy full SHA for 3ca8f94

Commits on Aug 8, 2023

• Remove option to use StreamExecutor Cloud TPU client in JAX

  Show description for 992ff1f
  

  skye

  authored and

  copybara-github

  committed
  992ff1f

  Copy full SHA for 992ff1f

• [XLA:Python] Remove unused xla_client_test.py parameter

  Show description for bdb9b6e
  

  skye

  authored and

  copybara-github

  committed
  bdb9b6e

  Copy full SHA for bdb9b6e

Commits on Aug 2, 2023

• PR #4666: Improve libtpu lockfile error messages

  Show description for 98efebe
  

  skye

  authored and

  copybara-github

  committed
  98efebe

  Copy full SHA for 98efebe

Commits on Jul 28, 2023

• [jax] Rewrite IfrtHelpers::xla_dynamic_shape to use more specific shape getters

  Show description for aa40dbe
  

  skye

  authored and

  copybara-github

  committed
  aa40dbe

  Copy full SHA for aa40dbe

• [IFRT] Rewrite PjRtArray::CopyToHostBuffer to use more specific shape getters

  Show description for f947e25
  

  skye

  authored and

  copybara-github

  committed
  f947e25

  Copy full SHA for f947e25

Commits on Jul 27, 2023

• [PJRT:C] Make PjRtCApiBuffer::has_dynamic_dimensions return false if unimplemented

  Show description for 8460f04
  

  skye

  authored and

  copybara-github

  committed
  8460f04

  Copy full SHA for 8460f04

• [XLA:Python] Use int64_t instead of ssize_t (part 2) to fix another Mac compiler error

  Show description for 69c7962
  

  skye

  authored and

  copybara-github

  committed
  69c7962

  Copy full SHA for 69c7962

Commits on Jul 26, 2023

• [XLA:Python] Use int64_t instead of ssize_t to fix Mac compiler errors (hopefully).

  Show description for d158d63
  

  skye

  authored and

  copybara-github

  committed
  d158d63

  Copy full SHA for d158d63

Commits on Jul 20, 2023

• [JAX:TPU] Don't ignore errors in xla_extension.get_c_api_client

  Show description for fb29e9f
  

  skye

  authored and

  copybara-github

  committed
  fb29e9f

  Copy full SHA for fb29e9f

Commits on Jul 19, 2023

• [XLA:Python] Replace dlpack call to PjRtBuffer::on_device_shape() with more specific call.

  Show description for 9441e26
  

  skye

  authored and

  copybara-github

  committed
  9441e26

  Copy full SHA for 9441e26

• [PJRT] Rewrite PjRtBuffer::ToLiteralSync to use new shape getters when possible.

  Show description for 97aec67
  

  skye

  authored and

  copybara-github

  committed
  97aec67

  Copy full SHA for 97aec67

• [PJRT:C] Add APIs for determining if there are dynamic dimensions.

  Show description for 8698196
  

  skye

  authored and

  copybara-github

  committed
  8698196

  Copy full SHA for 8698196

• [PJRT] Use new PjRtBuffer shape getters in xla/python

  Show description for d0cbed8
  

  skye

  authored and

  copybara-github

  committed
  d0cbed8

  Copy full SHA for d0cbed8

Pagination

Previous

Next