Commits

Branch selector

 main

User selector

sharadmv

Datepicker

All time

Commit History

Commits on Aug 3, 2023

• Update pyproject.toml

  sharadmv

  authored
  47fdca5

Commits on Aug 1, 2023

• [Pallas] Upstream pallas to JAX

  Show description for d258f8f
  

  sharadmv

  authored and

  The jax_triton Authors

  committed
  d258f8f

Commits on Jul 28, 2023

• Add Mosaic lowering for Pallas

  Show description for c6fc7b0
  

  sharadmv

  authored and

  The jax_triton Authors

  committed
  c6fc7b0

Commits on Jul 27, 2023

• * Test is flaky

  Show description for 46991ed
  

  sharadmv

  authored and

  The jax_triton Authors

  committed
  46991ed

Commits on Jul 17, 2023

• Separate out and add tests for Pallas indexing logic.

  Show description for 695afcf
  

  sharadmv

  authored and

  The jax_triton Authors

  committed
  695afcf

• Separate out and add tests for Pallas indexing logic.

  Show description for 163e1d4
  

  sharadmv

  authored and

  The jax_triton Authors

  committed
  163e1d4

Commits on Jul 14, 2023

• Update jax pin

  sharadmv

  committed
  765a353

• Update jax pin

  sharadmv

  committed
  bd0611e

Commits on Jun 27, 2023

• Add pl.when function

  Show description for b6897bc
  

  sharadmv

  authored and

  The jax_triton Authors

  committed
  b6897bc

Commits on Jun 14, 2023

• update jax version pin

  sharadmv

  committed
  8deec69

• Update tolerances for attn bwd test

  Show description for a89c4e1
  

  sharadmv

  authored and

  The jax_triton Authors

  committed
  a89c4e1

Commits on Jun 12, 2023

• Internal change

  Show description for 37c07df
  

  sharadmv

  authored and

  The jax_triton Authors

  committed
  37c07df

Commits on Jun 2, 2023

• update pins

  sharadmv

  committed
  dd62eaf

Commits on May 26, 2023

• Add pattern matching for while loop for fori_loop

  Show description for 35a8695
  sharadmv

  committed
  35a8695

Commits on May 25, 2023

• Fix bug in attn fwd

  sharadmv

  committed
  81264a1

Commits on May 24, 2023

• Move triton_kernel_call_lib to jaxlib

  Show description for a9b499d
  

  sharadmv

  authored and

  The jax_triton Authors

  committed
  a9b499d

Commits on May 23, 2023

• Pin versions in pyproject.toml

  sharadmv

  committed
  f160855

Commits on May 18, 2023

• Add support for ne

  sharadmv

  committed
  6a14a97

Commits on May 17, 2023

• Add support for causal attention to Pallas attention op

  sharadmv

  committed
  50cb554

• Add support for lowering fori_loop to Triton

  Show description for 79f26e4
  sharadmv

  committed
  79f26e4

Commits on May 16, 2023

• Separate out pallas registrations into their own file

  Show description for be19fe0
  

  sharadmv

  authored and

  The jax_triton Authors

  committed
  be19fe0

Commits on May 10, 2023

• Clean up pallas triton lowering

  Show description for e0fa6f5
  

  sharadmv

  authored and

  The jax_triton Authors

  committed
  e0fa6f5

Commits on Apr 14, 2023

• Internal change

  Show description for ac9b3ec
  

  sharadmv

  authored and

  The jax_triton Authors

  committed
  ac9b3ec

• Fix some whitespace

  Show description for 752b3e1
  

  sharadmv

  authored and

  The jax_triton Authors

  committed
  752b3e1

Commits on Apr 4, 2023

• Copybara import of the project:

  Show description for 03b9f9f
  

  sharadmv

  authored and

  The jax_triton Authors

  committed
  03b9f9f

Commits on Mar 21, 2023

• Rebase onto Triton-MLIR

  sharadmv

  committed
  de1073d

Commits on Mar 14, 2023

• Fix to bcast_in_dim and reshape

  sharadmv

  committed
  72bf2b2

Commits on Mar 7, 2023

• Fix jax_triton failures due to updated pe.debug_info signature

  Show description for 0678c96
  

  sharadmv

  authored and

  The jax_triton Authors

  committed
  0678c96

Commits on Feb 25, 2023

• Copybara import of the project:

  Show description for d80c479
  

  sharadmv

  authored and

  The jax_triton Authors

  committed
  d80c479

Commits on Feb 16, 2023

• Refactor GPU lowering into one file

  Show description for c42a887
  

  sharadmv

  authored and

  The jax_triton Authors

  committed
  c42a887

Commits on Feb 15, 2023

• Add atomic_cas to pallas

  sharadmv

  committed
  dfd85a7

Commits on Feb 10, 2023

• Bump version and add version to package

  sharadmv

  committed
  d3ad88f

• Bump version

  sharadmv

  committed
  6195df1

• Add option for XLA backward pass in flash attention

  sharadmv

  committed
  88b0f4d

Commits on Feb 9, 2023

• Handle not_mapped dimensions properly in pallas_call

  sharadmv

  committed
  2cf597d

Pagination

PreviousNext