nx icon indicating copy to clipboard operation
nx copied to clipboard
verified publisher icon elixir-nx

Update XLA

Open jonatanklosko opened this issue 6 months ago • 3 comments

Bumps XLA to recent revision.

We were still using the oldest API for defining custom calls and now a bunch of deprecation errors have been logged. I went ahead and updated to the latest typed FFI API, which also simplified the MLIR generation nicely.

Closes #1599.

jonatanklosko avatar Jun 16 '25 13:06 jonatanklosko

@polvalente all good!

jonatanklosko avatar Jun 16 '25 13:06 jonatanklosko

@polvalente all good!

Let me test this locally on CUDA later today before merging!

polvalente avatar Jun 16 '25 13:06 polvalente

Perfect!

jonatanklosko avatar Jun 16 '25 13:06 jonatanklosko

I've tried this PR with a 50xx card for a bit, and it looks good.

petrkozorezov avatar Jun 17 '25 18:06 petrkozorezov

@petrkozorezov fantastic, thank you for letting us know!

jonatanklosko avatar Jun 17 '25 18:06 jonatanklosko