pytorch
b52909d8 - [TensorExpr] Add python bindings for ArgValue class and TensorExprKernel constructor accepting custom lowerings. (#61385)

Commit
3 years ago
[TensorExpr] Add python bindings for ArgValue class and TensorExprKernel constructor accepting custom lowerings. (#61385) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/61385 The bindings coverage might be not full yet, but this already allows us to register custom lowerings from python. Test Plan: Imported from OSS Reviewed By: navahgar Differential Revision: D29623487 Pulled By: ZolotukhinM fbshipit-source-id: b97ee420a57fd887e204c021b9e098764b2ee232
Author
Mikhail Zolotukhin
Parents
Loading