Commits

Breadcrumbs

History for

onmain

User selector

All users

Datepicker

All time

Commit History

Commits on Feb 2, 2024

• [jax_triton] Only use side stream to do autotuning when doing graph capture

  Show description for 16636f9
  

  anlunx

  authored and

  jax authors

  committed
  16636f9

  Copy full SHA for 16636f9

Commits on Jan 29, 2024

• Make triton kernels compatible with command buffers

  Show description for 5e009f9
  

  anlunx

  authored and

  jax authors

  committed
  5e009f9

  Copy full SHA for 5e009f9

Commits on Jan 23, 2024

• The Triton autotuner ignores configs that use too much shmem

  Show description for 5761e39
  

  jax authors

  committed
  5761e39

  Copy full SHA for 5761e39

Commits on Jan 19, 2024

• [triton] Pass cluster_dims to TritonKernel and use cuLaunchKernel if size <= 1

  Show description for ab3c1b5
  

  jax authors

  committed
  ab3c1b5

  Copy full SHA for ab3c1b5

Commits on Oct 18, 2023

• [ROCm]: Add jax-triton support for ROCm

  

  Rahul Batra

  committed
  b4b97cd

  Copy full SHA for b4b97cd

Commits on Aug 16, 2023

• [jax_triton] Add user-specified name field to serialized format.

  Show description for 4ac2bdc
  

  chr1sj0nes

  authored and

  jax authors

  committed
  4ac2bdc

  Copy full SHA for 4ac2bdc

Commits on Jul 29, 2023

• [jax_triton] Add support for float scalar inputs.

  Show description for 714156d
  

  chr1sj0nes

  authored and

  jax authors

  committed
  714156d

  Copy full SHA for 714156d

Commits on Jul 5, 2023

• [jax_triton] Simplify auto-tuning code.

  Show description for 9935445
  

  chr1sj0nes

  authored and

  jax authors

  committed
  9935445

  Copy full SHA for 9935445

Commits on Jul 3, 2023

• [jax_triton] Split C++ only parts of Triton custom callback from Python parts.

  Show description for 31b862d
  

  chr1sj0nes

  authored and

  jax authors

  committed
  31b862d

  Copy full SHA for 31b862d

Loading