Commits

Branch selector

 main

User selector

ayaka14732

Datepicker

All time

Commit History

Commits on Dec 19, 2024

• [Pallas] Fix lowering tests for reduction ops

  Show description for 613a0cd
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  613a0cd

  Copy full SHA for 613a0cd

Commits on Dec 11, 2024

• [Pallas] Remove grid=1 in tests

  Show description for 13ce517
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  13ce517

  Copy full SHA for 13ce517

• [Pallas TPU] Add WeirdOp to TPU dialect and add lowering for lax.is_finite

  Show description for e88b578
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  e88b578

  Copy full SHA for e88b578

Commits on Dec 9, 2024

• [Pallas TPU] Improve lowerings for boolean comparison operations

  Show description for 9c98c0c
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  9c98c0c

  Copy full SHA for 9c98c0c

Commits on Dec 3, 2024

• [Pallas TPU] Enable test for jnp.logical_not because it's now supported

  Show description for 2dae81a
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  2dae81a

  Copy full SHA for 2dae81a

Commits on Nov 26, 2024

• [Pallas TPU] Better error message for lowering sp.broadcast_to_p

  Show description for dc11d40
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  dc11d40

  Copy full SHA for dc11d40

Commits on Nov 7, 2024

• [Pallas] Fix lowering tests for reduction ops

  Show description for 1a544b6
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  1a544b6

  Copy full SHA for 1a544b6

Commits on Nov 1, 2024

• [Pallas TPU] Add lowering for lax.nextafter

  Show description for f60b97c
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  f60b97c

  Copy full SHA for f60b97c

Commits on Oct 30, 2024

• [Pallas TPU] Add lowerings for scalar absi

  Show description for 8f96e90
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  8f96e90

  Copy full SHA for 8f96e90

Commits on Oct 29, 2024

• [Pallas] Add tests for jnp.logical_not

  Show description for a8d1048
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  a8d1048

  Copy full SHA for a8d1048

Commits on Oct 24, 2024

• [Pallas TPU] Add lowerings for scalar absf and rsqrt

  Show description for 5c61447
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  5c61447

  Copy full SHA for 5c61447

Commits on Oct 23, 2024

• [Pallas TPU] Fix OpsTest.test_elementwise test for bf16 inputs

  Show description for ea1fc65
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  ea1fc65

  Copy full SHA for ea1fc65

Commits on Oct 22, 2024

• [Pallas TPU] Fix lowering for jnp.remainder

  Show description for c60bafc
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  c60bafc

  Copy full SHA for c60bafc

• [Pallas TPU] Add lowerings for bf16 jnp.ceil and jnp.floor in TPU v6+

  Show description for 2b7b074
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  2b7b074

  Copy full SHA for 2b7b074

Commits on Oct 18, 2024

• [Pallas TPU] Use new MLIR op names

  Show description for 884f1dc
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  884f1dc

  Copy full SHA for 884f1dc

Commits on Oct 15, 2024

• [Pallas TPU] Fix boolean comparison

  Show description for 5ac2076
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  5ac2076

  Copy full SHA for 5ac2076

Commits on Oct 14, 2024

• [Pallas TPU] Add lowerings for scalar sin, cos, tan and tanh

  Show description for bfc3d3c
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  bfc3d3c

  Copy full SHA for bfc3d3c

Commits on Oct 11, 2024

• [Pallas TPU] Add lowering for scalar jnp.log1p

  Show description for 633cb31
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  633cb31

  Copy full SHA for 633cb31

Commits on Oct 10, 2024

• [Pallas] Add tests for scalar elementwise operations

  Show description for 3bd8ca4
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  3bd8ca4

  Copy full SHA for 3bd8ca4

Commits on Oct 9, 2024

• [Pallas TPU] Add lowerings for lax.population_count_p and lax.clz_p

  Show description for 3fc4ba2
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  3fc4ba2

  Copy full SHA for 3fc4ba2

• [Pallas TPU] Raise NotImplementedError when casting to 64-bit integer

  Show description for df6a0fb
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  df6a0fb

  Copy full SHA for df6a0fb

• [Pallas TPU] Fix comparison lowering for unsigned integers

  Show description for 77613f2
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  77613f2

  Copy full SHA for 77613f2

Commits on Oct 8, 2024

• [Pallas] Simplify OpsTest by skipping 64-bit tests on 32-bit environments

  Show description for 6a958b9
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  6a958b9

  Copy full SHA for 6a958b9

Commits on Oct 3, 2024

• [Pallas TPU] Add lowering for lax.ceil_p

  Show description for cb2e0e2
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  cb2e0e2

  Copy full SHA for cb2e0e2

• [Pallas] [Docs] Replace full urls with label-based cross references

  Show description for e79d77a
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  e79d77a

  Copy full SHA for e79d77a

• [Docs] Add docstring for RoundingMethod

  Show description for ad78147
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  ad78147

  Copy full SHA for ad78147

• [Pallas TPU] Add lowering for lax.floor_p

  Show description for b5ce445
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  b5ce445

  Copy full SHA for b5ce445

Commits on Oct 1, 2024

• [Pallas TPU] Add lowering for lax.tan_p

  Show description for e361868
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  e361868

  Copy full SHA for e361868

Commits on Sep 30, 2024

• [Pallas TPU] Add lowering for lax.cos_p

  Show description for a24420e
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  a24420e

  Copy full SHA for a24420e

• [Pallas TPU] Consolidate OpsExtraTest into OpsTest

  Show description for 23ce5a1
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  23ce5a1

  Copy full SHA for 23ce5a1

Commits on Sep 27, 2024

• [Pallas TPU] Add a note in the Pallas Quickstart documentation about the instructions of running the existing example on TPU

  Show description for ab4590c
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  ab4590c

  Copy full SHA for ab4590c

Commits on Sep 25, 2024

• [Pallas TPU] Improve error message when trying to store a scalar to VMEM

  Show description for 02cfaa8
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  02cfaa8

  Copy full SHA for 02cfaa8

Commits on Sep 23, 2024

• [Pallas] Simplify sign and erf_inv tests

  Show description for 93203c7
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  93203c7

  Copy full SHA for 93203c7

• [Pallas] Skip atomic_cas and atomic_counter tests on GPU in 64-bit mode

  Show description for b6fe793
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  b6fe793

  Copy full SHA for b6fe793

Commits on Sep 20, 2024

• [Pallas GPU] Enable Pallas OpsExtraTest in 64-bit mode

  Show description for d63afd8
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  d63afd8

  Copy full SHA for d63afd8

Pagination

Previous

Next