search

jax-ml / jax-triton

jax-triton contains integrations between JAX and OpenAI Triton
Apache License 2.0
350 stars 40 forks source link

Removed the code path for `ttir` being an `ir.Module` #297

Closed copybara-service[bot] closed 2 months ago

copybara-service[bot] commented 2 months ago

Removed the code path for ttir being an ir.Module

This was only necessary when Pallas GPU was using jax_triton for lowering.