Commits

Breadcrumbs

History for

onmain

User selector

All users

Datepicker

All time

Commit History

Commits on Nov 21, 2024

• Reverts c04aec9

  Show description for e72b449
  

  naummo

  authored and

  Google-ML-Automation

  committed
  e72b449

  Copy full SHA for e72b449

• [Mosaic TPU] Add bound check for general vector store op.

  Show description for 869a533
  

  bythew3i

  authored and

  Google-ML-Automation

  committed
  869a533

  Copy full SHA for 869a533

Commits on Nov 20, 2024

• Mention python 3.13 in docs & package metadata

  jakevdp

  committed
  a4266b5

  Copy full SHA for a4266b5

• [Mosaic] Add tpu.log verification on SC

  Show description for 6c291d6
  

  naummo

  authored and

  Google-ML-Automation

  committed
  6c291d6

  Copy full SHA for 6c291d6

Commits on Nov 19, 2024

• [Mosaic] Extend tpu.sem_signal with subcore_id

  Show description for c04aec9
  

  naummo

  authored and

  Google-ML-Automation

  committed
  c04aec9

  Copy full SHA for c04aec9

• [Mosaic] Add target core type parameter to tpu.sem_signal

  Show description for 0d36b0b
  

  naummo

  authored and

  Google-ML-Automation

  committed
  0d36b0b

  Copy full SHA for 0d36b0b

• [Mosaic TPU] Add general tpu.vector_store and support masked store.

  Show description for 6c31efa
  

  bythew3i

  authored and

  Google-ML-Automation

  committed
  6c31efa

  Copy full SHA for 6c31efa

• Implement lax.pad in Pallas.

  Show description for d397dd9
  Google-ML-Automation

  committed
  d397dd9

  Copy full SHA for d397dd9

• [Mosaic TPU] Support relayout for mask vector

  Show description for 0fe77bc
  

  bythew3i

  authored and

  Google-ML-Automation

  committed
  0fe77bc

  Copy full SHA for 0fe77bc

Commits on Nov 18, 2024

• Add a GPU implementation of lax.linalg.eig.

  Show description for ccb3317
  

  dfm

  authored and

  Google-ML-Automation

  committed
  ccb3317

  Copy full SHA for ccb3317

Commits on Nov 15, 2024

• [Mosaic TPU] Support 1D concat: set implicit_dim to kSecondMinor to treat 1D (N,) as (1, N) and then tile it as (1, 128)

  Show description for 1471702
  Google-ML-Automation

  committed
  1471702

  Copy full SHA for 1471702

Commits on Nov 14, 2024

• [Mosaic][TPU] Omit short circuiting of relayout (we should always relayout!) and implement product mismatch case for where we relayout from replicated to offset, and the number of vregs changes.

  Show description for a8464ce
  Google-ML-Automation

  committed
  a8464ce

  Copy full SHA for a8464ce

Commits on Nov 12, 2024

• [Mosaic] Add parameter names to tpu.sem_signal and add tests

  Show description for c32db46
  

  naummo

  authored and

  Google-ML-Automation

  committed
  c32db46

  Copy full SHA for c32db46

• [mosaic_gpu] The profiler now uses FFI calls for creating events and computing elapsed time

  Show description for d304025
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  d304025

  Copy full SHA for d304025

• [Mosaic] Fix mask creation for packed sublanes

  Show description for 1221da8
  Google-ML-Automation

  committed
  1221da8

  Copy full SHA for 1221da8

• Fix overflow error in GPU batched linear algebra kernels.

  Show description for 21e98b5
  

  dfm

  authored and

  Google-ML-Automation

  committed
  21e98b5

  Copy full SHA for 21e98b5

• [Mosaic TPU] Support dynamic DMA and ref slice on the 2nd minor when memref is untiled

  Show description for 38d062d
  

  bythew3i

  authored and

  Google-ML-Automation

  committed
  38d062d

  Copy full SHA for 38d062d

Commits on Nov 11, 2024

• [Mosaic GPU] Add base_pointer argument to InitializeBarrierOp.

  Show description for da89c9e
  

  bchetioui

  authored and

  Google-ML-Automation

  committed
  da89c9e

  Copy full SHA for da89c9e

Commits on Nov 8, 2024

• [MOSAIC:GPU] Add async_load, async_store, and supporting attributes to the MLIR Mosaic GPU Dialect.

  Show description for d833066
  

  dimitar-asenov

  authored and

  Google-ML-Automation

  committed
  d833066

  Copy full SHA for d833066

• Merge pull request jax-ml#24778 from cainmagi:fix-pr-23852

  Show description for 4d1a126
  Google-ML-Automation

  committed
  4d1a126

  Copy full SHA for 4d1a126

• [Mosaic GPU] Make sure to free the cloned MLIR module when debugging

  Show description for ce3826d
  

  apaszke

  authored and

  Google-ML-Automation

  committed
  ce3826d

  Copy full SHA for ce3826d

• (follow-up of PR jax-ml#23852) add missing typename keyword to work with gcc

  Show description for 218f763
  cainmagi

  authored
  218f763

  Copy full SHA for 218f763

Commits on Nov 7, 2024

• [Mosaic] Fix handling of i1 splat constants

  Show description for 04a6652
  

  tlongeri

  authored and

  Google-ML-Automation

  committed
  04a6652

  Copy full SHA for 04a6652

• [Mosaic] Fix canonicalize_extract op name.

  Show description for 8b7bcad
  

  WindQAQ

  authored and

  Google-ML-Automation

  committed
  8b7bcad

  Copy full SHA for 8b7bcad

Commits on Nov 6, 2024

• [Mosaic] Verify that tpu.sem_wait semaphore rank is zero

  Show description for 3df204a
  

  naummo

  authored and

  Google-ML-Automation

  committed
  3df204a

  Copy full SHA for 3df204a

Commits on Nov 5, 2024

• Include mpmath as a bazel dependency of lax_test.

  Show description for ea1e879
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  ea1e879

  Copy full SHA for ea1e879

• [mosaic_gpu] Check the return code of gpuEventCreate and gpuEventDestroy

  Show description for 34b4787
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  34b4787

  Copy full SHA for 34b4787

• [Mosaic GPU] Ensure that the dialect module can be loaded successfully.

  Show description for 63e59c5
  

  bchetioui

  authored and

  Google-ML-Automation

  committed
  63e59c5

  Copy full SHA for 63e59c5

Commits on Oct 31, 2024

• [Mosaic] Add extension files for infer/apply vector layout.

  Show description for 8296f6e
  

  18praveenb

  authored and

  Google-ML-Automation

  committed
  8296f6e

  Copy full SHA for 8296f6e

• [Mosaic] Fix some imports.

  Show description for 7d9f565
  

  18praveenb

  authored and

  Google-ML-Automation

  committed
  7d9f565

  Copy full SHA for 7d9f565

• [Mosaic GPU] Add Python bindings for the Mosaic GPU MLIR dialect.

  Show description for c708a04
  

  bchetioui

  authored and

  Google-ML-Automation

  committed
  c708a04

  Copy full SHA for c708a04

• [MOSAIC:GPU] Extend the mosaic mlir dialect with fragmented layouts.

  Show description for 7d504cd
  

  dimitar-asenov

  authored and

  Google-ML-Automation

  committed
  7d504cd

  Copy full SHA for 7d504cd

• [Mosaic] Extend tpu matmulop to have dimension dims. Add support for batching and simple transposition.

  Show description for 5aeffde
  Google-ML-Automation

  committed
  5aeffde

  Copy full SHA for 5aeffde

Commits on Oct 30, 2024

• [Mosaic] Add the core type enum

  Show description for 242e663
  

  naummo

  authored and

  Google-ML-Automation

  committed
  242e663

  Copy full SHA for 242e663

• [Fix] Put * packing into reshape no-op condition (Bug in my original CL)

  Show description for 99ea4c1
  Google-ML-Automation

  committed
  99ea4c1

  Copy full SHA for 99ea4c1

Pagination

Previous

Next