Skip to content

v0.3.2

Choose a tag to compare

@oulgen oulgen released this 13 Mar 21:33
59927cc

What's Changed

  • [pallas-tpu] Enable test_geglu, test_swiglu on Pallas; fix softmax/kl_div/gdn xfails by @v0i0 in #1637
  • [pallas-tpu] make dot work: use jnp.matmul and f32 accumulator by @v0i0 in #1647
  • [pallas-tpu] Enable broadcasting and control flow tests for Pallas backend by @v0i0 in #1643
  • Use ephemeral Triton cache during autotuning to eliminate loser artifacts by @fulvius31 in #1649
  • Try public bazel cache for torch_tpu build by @v0i0 in #1658
  • fix cudagraph grouped gemm fail in benchmarks by @v0i0 in #1662
  • Use matmul fwd direclty in autograd for performance by @tianrengao in #1045
  • Recreate foreign SymInts in helion's own ShapeEnv by @gmagogsfm in #1671

Full Changelog: v0.3.1...v0.3.2