pytorch
c9ab384a - [TensorExpr] Add `CodeGen::call_raw` method. (#55113)

Commit
3 years ago
[TensorExpr] Add `CodeGen::call_raw` method. (#55113) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55113 The new method allows to pass input and output arguments by `void*` pointers instead of CallArgs. That helps to reduce the invocation overhead. Currently this is only supported in LLVM codegen. Differential Revision: D27487549 Test Plan: Imported from OSS Reviewed By: bertmaher Pulled By: ZolotukhinM fbshipit-source-id: d8f3d92262cde1c155beefb629454370d9af2f89
Author
Mikhail Zolotukhin
Parents
Loading