pytorch
4e15a6f4 - [TensorExpr] Switch Exprs and Stmt from kernel-arena to shared_ptr. (#63216)

Commit
4 years ago
[TensorExpr] Switch Exprs and Stmt from kernel-arena to shared_ptr. (#63216) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/63216 Currently there are three classes managed by KernelArena: Expr, Stmt, and Tensor (and derived classes). KernelArena has been a long standing painpoint for NNC devs and we're moving away from that memory management model to ref-count based memory model (using shared_ptr). This commit switches Expr and Stmt to shared_ptr and is the biggest change in this transition. Later commits will detach Tensor from KernelArena and kill the arena + scope altogether. Differential Revision: D30353195 D30353195 Test Plan: Imported from OSS Reviewed By: navahgar Pulled By: ZolotukhinM fbshipit-source-id: 9575225ada3d0fb65087ae40435f3dfea4792cae
Author
Mikhail Zolotukhin
Parents
Loading