Skip to content

Conversation

dlibenzi
Copy link
Collaborator

Exposes XLA builder capabilities to the Python layer.
This allows faster development cycle when prototyping ops, and allow users to define pytorch operations in terms of raw HLO (via XLA builder APIs), without having to build new binaries.
There are still XLA builder ops missing, and documentation (API final shape under development).

@dlibenzi dlibenzi requested review from JackCaoG and ailzhang May 25, 2020 16:36
@dlibenzi dlibenzi force-pushed the xla_user_ops branch 12 times, most recently from eeb15f5 to 6fc4573 Compare May 26, 2020 18:52
Copy link
Collaborator

@JackCaoG JackCaoG left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Wow.. it is a huge diff and takes a while to go through. Is some of them auto generated 🤣? Thanks for making this change!

@dlibenzi
Copy link
Collaborator Author

Wow.. it is a huge diff and takes a while to go through. Is some of them auto generated 🤣? Thanks for making this change!

No, but the interesting bits are few 😉
The bulk (xla_op_builder.cpp and xla_builder.py) are just pretty dumb mapping of Python functions to XLA builder generation.

@dlibenzi dlibenzi merged commit f0b6295 into master May 27, 2020
@dlibenzi dlibenzi deleted the xla_user_ops branch May 27, 2020 00:58
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.

2 participants