Commits

Branch selector

 main

User selector

superbobry

Datepicker

All time

Commit History

Commits on Feb 15, 2024

• Check if the Triton dialect bindings are available in lib/triton.py

  Show description for b4c8b0e
  

  superbobry

  authored and

  jax authors

  committed
  b4c8b0e

  Copy full SHA for b4c8b0e

Commits on Feb 14, 2024

• Inlined triton.compat

  Show description for 8814362
  

  superbobry

  authored and

  jax authors

  committed
  8814362

  Copy full SHA for 8814362

Commits on Feb 13, 2024

• Dropped the Triton* prefix from class names in triton/lowering.py

  Show description for 0736dd7
  

  superbobry

  authored and

  jax authors

  committed
  0736dd7

  Copy full SHA for 0736dd7

• Changed lower_jaxpr_to_triton_ir to use ir.Value instead of triton.compat APIs

  Show description for cee8bf7
  

  superbobry

  authored and

  jax authors

  committed
  cee8bf7

  Copy full SHA for cee8bf7

• Added ir.Value-based versions of load and store in triton.compat

  Show description for 6a7d1dc
  

  superbobry

  authored and

  jax authors

  committed
  6a7d1dc

  Copy full SHA for 6a7d1dc

• Added ir.Value-based versions of some triton.compat APIs

  Show description for 0ddf371
  

  superbobry

  authored and

  jax authors

  committed
  0ddf371

  Copy full SHA for 0ddf371

Commits on Feb 12, 2024

• Updated the Triton lowering of jnp.transpose

  Show description for 7dd887d
  

  superbobry

  authored and

  jax authors

  committed
  7dd887d

  Copy full SHA for 7dd887d

Commits on Feb 9, 2024

• Moved atomic_* operations out of the Triton compatibility layer

  Show description for 8a8601f
  

  superbobry

  authored and

  jax authors

  committed
  8a8601f

  Copy full SHA for 8a8601f

Commits on Feb 8, 2024

• Inlined tl.core._to_tensor

  Show description for 4c505f8
  

  superbobry

  authored and

  jax authors

  committed
  4c505f8

  Copy full SHA for 4c505f8

Commits on Feb 7, 2024

• _triton_ext no longer links in MLIR C APIs

  Show description for 5e2e609
  

  superbobry

  authored and

  jax authors

  committed
  5e2e609

  Copy full SHA for 5e2e609

• Add MLIR_CAPI_EXPORTED to Triton dialect C bindings

  Show description for 01b995e
  

  superbobry

  authored and

  jax authors

  committed
  01b995e

  Copy full SHA for 01b995e

Commits on Feb 5, 2024

• Fixed a typo in min/max Triton lowering rules

  Show description for 9e94e6e
  

  superbobry

  authored and

  jax authors

  committed
  9e94e6e

  Copy full SHA for 9e94e6e

• Removed unnecessary calls to jax_triton.get_triton_type

  Show description for 5c5e8f0
  

  superbobry

  authored and

  jax authors

  committed
  5c5e8f0

  Copy full SHA for 5c5e8f0

• Inlined all builder.create_* calls

  Show description for 0596c80
  

  superbobry

  authored and

  jax authors

  committed
  0596c80

  Copy full SHA for 0596c80

• compile_ttir_to_ptx_inplace no longer requires an IR context

  Show description for c0c6a45
  

  superbobry

  authored and

  jax authors

  committed
  c0c6a45

  Copy full SHA for c0c6a45

• Migrated tl.semantic.cast to lower directly to MLIR

  Show description for c77d45d
  

  superbobry

  authored and

  jax authors

  committed
  c77d45d

  Copy full SHA for c77d45d

Commits on Feb 3, 2024

• Inlined some Triton-specific abstractions

  Show description for 06d3280
  

  superbobry

  authored and

  jax authors

  committed
  06d3280

  Copy full SHA for 06d3280

Commits on Feb 2, 2024

• Removed wrap_with_builder and tensor.to from the Triton compatibility layer

  Show description for cda40ec
  

  superbobry

  authored and

  jax authors

  committed
  cda40ec

  Copy full SHA for cda40ec

• Migrated dot to lower directly to Triton IR

  Show description for 28eff4f
  

  superbobry

  authored and

  jax authors

  committed
  28eff4f

  Copy full SHA for 28eff4f

• Migrated store/load to lower directly to Triton IR

  Show description for 5867a05
  

  superbobry

  authored and

  jax authors

  committed
  5867a05

  Copy full SHA for 5867a05

• Added a custom lowering rule for pow which special-cases weak dtypes

  Show description for e1ea936
  

  superbobry

  authored and

  jax authors

  committed
  e1ea936

  Copy full SHA for e1ea936

• Fixed a few ir.Value type-tensor dtype mismatches in the Pallas lowering code on GPU

  Show description for d9f42c5
  

  superbobry

  authored and

  jax authors

  committed
  d9f42c5

  Copy full SHA for d9f42c5

Commits on Jan 30, 2024

• Temporarily switch triton.compat to use Triton APIs for math and semantic operations

  Show description for 9e76e38
  

  superbobry

  authored and

  jax authors

  committed
  9e76e38

  Copy full SHA for 9e76e38

Commits on Jan 29, 2024

• Migrated remaining operations from the math namespace to lower directly to Triton IR

  Show description for fad3e74
  

  superbobry

  authored and

  jax authors

  committed
  fad3e74

  Copy full SHA for fad3e74

• Migrated atomic operations to lower directly to Triton IR

  Show description for 07f8f70
  

  superbobry

  authored and

  jax authors

  committed
  07f8f70

  Copy full SHA for 07f8f70

• Slice.from_slice now works for slices with a negative start index

  Show description for 3cea57d
  superbobry

  committed
  3cea57d

  Copy full SHA for 3cea57d

• Replaced most usages of abc.ABC with util.StrictABC

  Show description for 078bb00
  superbobry

  committed
  078bb00

  Copy full SHA for 078bb00

Commits on Jan 26, 2024

• Ported a subset of binary operations to lower directly to Triton IR

  Show description for cc5f565
  

  superbobry

  authored and

  jax authors

  committed
  cc5f565

  Copy full SHA for cc5f565

• Fixed a bug in _reduction_lowering

  Show description for cb7a32a
  

  superbobry

  authored and

  jax authors

  committed
  cb7a32a

  Copy full SHA for cb7a32a

• compat.tensor __*__ methods no longer do implicit broadcasting

  Show description for 273cb27
  

  superbobry

  authored and

  jax authors

  committed
  273cb27

  Copy full SHA for 273cb27

• Fixed a typo in Pallas GPU lowering

  Show description for f34bcc3
  

  superbobry

  authored and

  jax authors

  committed
  f34bcc3

  Copy full SHA for f34bcc3

Commits on Jan 24, 2024

• Lower a subset of math primitives directly to Triton IR

  Show description for f15cad4
  

  superbobry

  authored and

  jax authors

  committed
  f15cad4

  Copy full SHA for f15cad4

Commits on Jan 22, 2024

• Dedupe shardings before passing them to _get_and_check_device_assignment

  Show description for 46f796b
  

  superbobry

  authored and

  jax authors

  committed
  46f796b

  Copy full SHA for 46f796b

• DOC Add a noop implementation of the legacy directive to fix the build

  superbobry

  committed
  4196bf1

  Copy full SHA for 4196bf1

Commits on Jan 16, 2024

• Include Triton files into the jaxlib wheel

  Show description for 1e9f96a
  superbobry

  committed
  1e9f96a

  Copy full SHA for 1e9f96a

Pagination

Previous

Next