Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Build against openxla XLA repo #393

Merged
merged 464 commits into from
Jun 8, 2024
Merged

Build against openxla XLA repo #393

merged 464 commits into from
Jun 8, 2024

Conversation

joelberkeley
Copy link
Owner

@joelberkeley joelberkeley commented Mar 3, 2024

BEFORE MERGE

  • check pack install actually works including dynamic libs (either create a toy project and add that to pack, or use a fork of pack-db)
  • delete releases and tags

commit message

migrate to OpenXLA

  • use PJRT to compile and execute graph
  • can now choose the runtime device, via wrapped PJRT plugins
  • allow users to create and use custom PJRT plugins
  • remove source dependency on TensorFlow
  • build all binaries in spidr, rather than using elixir-nx
  • add build (not test) for CUDA in CI
  • automatically install support libraries to idris path (no longer global for user)
  • add tutorial on spidr internals

@joelberkeley joelberkeley marked this pull request as ready for review June 7, 2024 00:51
@joelberkeley joelberkeley merged commit 7926b1a into master Jun 8, 2024
7 of 11 checks passed
@joelberkeley joelberkeley deleted the xla-build branch June 8, 2024 00:16
joelberkeley added a commit that referenced this pull request Jun 14, 2024
* use PJRT to compile and execute graph
* allow users to choose runtime device, via wrapped PJRT plugins
* allow users to create and use custom PJRT plugins
* remove source dependency on TensorFlow
* build all binaries in spidr, rather than using elixir-nx
* build (not test) spidr for CUDA in CI
* automatically install support libraries to idris path (not global for user)
* add tutorial on spidr internals
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant