pytorch
dedaf4fa - Reland: [TensorExpr] Add methods for inspecting generated code in `TensorExprKernel`. (#57560)

Commit
3 years ago
Reland: [TensorExpr] Add methods for inspecting generated code in `TensorExprKernel`. (#57560) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/57560 The new methods allow to peak into bufferArgs which describe parameters that codegen expects. This description includes info whether a given parameter is a scalar var or a buffer and in case it's a buffer allows to get the corresponding `Buf*` pointer from which we could get the expected sizes. Relanding #57074 which was reverted because I forgot to guard a new test with `ifdef LLVM`. Test Plan: Imported from OSS Reviewed By: bertmaher Differential Revision: D28199048 Pulled By: ZolotukhinM fbshipit-source-id: 636e838e7e242a3c63e97ec453b8fae9b6380231
Author
Mikhail Zolotukhin
Parents
Loading