Align the custom_call implementation in mlir and hlo_helpers. - Githubissues

jax-ml / jax-triton

jax-triton contains integrations between JAX and OpenAI Triton

Apache License 2.0

340 stars 36 forks source link

Align the custom_call implementation in mlir and hlo_helpers. #227

Closed copybara-service[bot] closed 1 year ago

copybara-service[bot] commented 1 year ago

Align the custom_call implementation in mlir and hlo_helpers.