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

Add external calling functionality #187

Open
wants to merge 294 commits into
base: pytorch_fusion
Choose a base branch
from

Conversation

bwasti
Copy link
Collaborator

@bwasti bwasti commented Feb 21, 2020

This PR adds the ability to register external functions directly into the IR.

An example, matmul, is used to show how this registration is done.

This will bootstrap operators that are currently (or always) hard to generate. It also enables composability with other specialized compilers (such as FBGEMM).

ZolotukhinM and others added 19 commits February 24, 2020 15:54
…ts. (pytorch#194)

All `test_*` functions are now moved into a test-class (with no changes to them).
* Add rand benchmark.
* Add an option to disable texpr fuser.
…remainder (pytorch#198)

* Add the cast_float, backward ops and also fix the remainder

* fix the conflict

* change expr to exprhandle

* formatting

* fix the linter
* Fix some ir printer bugs

* also true_stmt
…rch#142)

* Enable axis splitting and GPU grid binding with variable shapes

* farwell ExprStmt, we hardly knew ye
* Add workflow.md.

* Remove the suggestions from the doc.

* Add language reference.

* Add language reference.

* Address some of the comments.
* Adding bitwise integer ops: &,^,<<, >>
* Add the cast_float, backward ops and also fix the remainder

fix the conflict

change expr to exprhandle

formatting

fix the linter

add the type_as support

* fix the threshold failure
* Aten op: where
This require a helper function which does promote types for the condition expression.
* LLVM codgen for fmod, remainder
* fix testATengeInt
@bwasti bwasti changed the title [needs rebase] Add external calling functionality Add external calling functionality Feb 27, 2020
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.

10 participants