Commits

Breadcrumbs

History for

onmaster

User selector

All users

Datepicker

All time

Commit History

Commits on Dec 23, 2024

• [Mosaic:TPU] In infer ext rule, avoid assigning offsets outside of dst first tile

  Show description for 4452960
  

  tlongeri

  authored and

  Google-ML-Automation

  committed
  4452960

  Copy full SHA for 4452960

• Switch mlir bindings from pybind11 to nanobind

  Show description for b8091a4
  Google-ML-Automation

  committed
  b8091a4

  Copy full SHA for b8091a4

• [Mosaic:TPU] Vreg-slice-aligned offset changes with scratch retiling

  Show description for 3c79b98
  

  tlongeri

  authored and

  Google-ML-Automation

  committed
  3c79b98

  Copy full SHA for 3c79b98

• Use the right include for gmock and gtest

  Show description for 68ec202
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  68ec202

  Copy full SHA for 68ec202

• [mosaic_gpu] Include Mosaic GPU dialect fiels into jaxlib

  superbobry

  committed
  8987867

  Copy full SHA for 8987867

Commits on Dec 20, 2024

• [Mosaic:TPU] Roll forward of cl/708011538 (expanded trunc support), minus changes in infer-vector-layout

  Show description for 7ecc947
  

  tlongeri

  authored and

  Google-ML-Automation

  committed
  7ecc947

  Copy full SHA for 7ecc947

• Migrate _mlir Python binding target to nanobind.

  Show description for 0ff3f14
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  0ff3f14

  Copy full SHA for 0ff3f14

• [Mosaic:TPU] Roll back cl/708011538 and cl/708112341

  Show description for 05f3a70
  

  tlongeri

  authored and

  Google-ML-Automation

  committed
  05f3a70

  Copy full SHA for 05f3a70

• [Mosaic TPU] Add relayout-insertion pass and support bitwidth change for i1 vector relayout

  Show description for 2faf540
  

  bythew3i

  authored and

  Google-ML-Automation

  committed
  2faf540

  Copy full SHA for 2faf540

• [Mosaic:TPU] Fix trunc infer rule after cl/708011538

  Show description for 8b02884
  

  tlongeri

  authored and

  Google-ML-Automation

  committed
  8b02884

  Copy full SHA for 8b02884

Commits on Dec 19, 2024

• [Mosaic] Extend macros to handle parentheses.

  Show description for 60ebde8
  

  WindQAQ

  authored and

  Google-ML-Automation

  committed
  60ebde8

  Copy full SHA for 60ebde8

• [Mosaic] Remove TODOs that are already addressed or obsolete.

  Show description for 77f3c11
  

  WindQAQ

  authored and

  Google-ML-Automation

  committed
  77f3c11

  Copy full SHA for 77f3c11

• Use common set of build options when building jaxlib+plugin artifacts together

  Show description for 6b096b0
  

  nitins17

  authored and

  Google-ML-Automation

  committed
  6b096b0

  Copy full SHA for 6b096b0

• [Mosaic:TPU] For trunc, expand supported tilings, offsets and bitwidths

  Show description for 307c8d3
  

  tlongeri

  authored and

  Google-ML-Automation

  committed
  307c8d3

  Copy full SHA for 307c8d3

• Migrate shardy dialect extension to nanobind.

  Show description for 9c3365f
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  9c3365f

  Copy full SHA for 9c3365f

• [Mosaic GPU] Commit to using Vectors everywhere (and no Tensors).

  Show description for 3915f4a
  

  bchetioui

  authored and

  Google-ML-Automation

  committed
  3915f4a

  Copy full SHA for 3915f4a

• [jaxlib] Added a missing pytype_dep to :_triton_ext

  Show description for c4fae4a
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  c4fae4a

  Copy full SHA for c4fae4a

• [Mosaic GPU] Replace the dialect's layout enum with layouts holding the proper

  Show description for 66ad208
  

  bchetioui

  authored and

  Google-ML-Automation

  committed
  66ad208

  Copy full SHA for 66ad208

• [Mosaic:TPU][NFC] Small cleanup of extui rule in apply-vector-layout

  Show description for 8188c57
  

  tlongeri

  authored and

  Google-ML-Automation

  committed
  8188c57

  Copy full SHA for 8188c57

• [Mosaic TPU] Support i32 vector multi reduction except cross lane.

  Show description for 3a5c4da
  

  bythew3i

  authored and

  Google-ML-Automation

  committed
  3a5c4da

  Copy full SHA for 3a5c4da

• [Mosaic] Improve error verbosity of tpu.memref_slice verification

  Show description for 6bcec91
  

  naummo

  authored and

  Google-ML-Automation

  committed
  6bcec91

  Copy full SHA for 6bcec91

Commits on Dec 18, 2024

• [Mosaic] Verify that the target IDs are provided in remote DMAs

  Show description for de359f5
  

  naummo

  authored and

  Google-ML-Automation

  committed
  de359f5

  Copy full SHA for de359f5

• [Mosaic:TPU][NFC] Delete unused functions

  Show description for 13e721a
  

  tlongeri

  authored and

  Google-ML-Automation

  committed
  13e721a

  Copy full SHA for 13e721a

• [Mosaic TPU] Support direct cast i8 vector to mask

  Show description for bf692ef
  

  bythew3i

  authored and

  Google-ML-Automation

  committed
  bf692ef

  Copy full SHA for bf692ef

• Merge pull request jax-ml#25569 from hawkinsp:numpyver

  Show description for 464e5a2
  Google-ML-Automation

  committed
  464e5a2

  Copy full SHA for 464e5a2

• Migrate mhlo dialect extension to nanobind.

  Show description for 3f24dfd
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  3f24dfd

  Copy full SHA for 3f24dfd

• [Mosaic TPU] Add support for bf16 second minor reductions in TPUv6

  Show description for 6edfe9e
  

  apaszke

  authored and

  Google-ML-Automation

  committed
  6edfe9e

  Copy full SHA for 6edfe9e

• Migrate StableHLO Python extension to nanobind.

  Show description for 3d54d03
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  3d54d03

  Copy full SHA for 3d54d03

• Increase the minimum NumPy version to v1.25.

  Show description for ee45718
  hawkinsp

  committed
  ee45718

  Copy full SHA for ee45718

• [Mosaic:TPU] Allow null parts for tpu.pack_subelements, meaning "don't care"

  Show description for dc0b774
  

  tlongeri

  authored and

  Google-ML-Automation

  committed
  dc0b774

  Copy full SHA for dc0b774

• [Mosaic:TPU] Fix bug after cl/707025084

  Show description for f9737b9
  

  tlongeri

  authored and

  Google-ML-Automation

  committed
  f9737b9

  Copy full SHA for f9737b9

Commits on Dec 17, 2024

• [Mosaic TPU] Add support for the interleaved pack format to tpu.unpack_subelements

  Show description for 4911a39
  

  apaszke

  authored and

  Google-ML-Automation

  committed
  4911a39

  Copy full SHA for 4911a39

• [Mosaic GPU] Add end-to-end lowering example for a pointwise kernel using the dialect and layout inference.

  Show description for 36b12d5
  

  bchetioui

  authored and

  Google-ML-Automation

  committed
  36b12d5

  Copy full SHA for 36b12d5

• [Mosaic:TPU][NFC] use_implicit_shape instead of Reshape + don't create unused constant

  Show description for c6e6b11
  

  tlongeri

  authored and

  Google-ML-Automation

  committed
  c6e6b11

  Copy full SHA for c6e6b11

• [Mosaic:TPU][NFC] Clean up local variable

  Show description for 7e6c52d
  

  tlongeri

  authored and

  Google-ML-Automation

  committed
  7e6c52d

  Copy full SHA for 7e6c52d

Pagination

Previous

Next