projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 527348a) | patch
[tensorexpr] Allocate intermediate buffers at compile time (#64227)
authorHui Guo <huiguo@fb.com>
Wed, 8 Sep 2021 22:30:59 +0000 (15:30 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Wed, 8 Sep 2021 22:34:44 +0000 (15:34 -0700)
commit5c27a580ecae83ce9aa253c6a4b0638aeb00d0f3
treec2482008b1dc0745ae71660d614a8b165b854d53tree | snapshot
parent527348a6fe934dcfeeb88352e0cb9e263c76b78dcommit | diff
[tensorexpr] Allocate intermediate buffers at compile time (#64227)

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/64227

Test Plan: Imported from OSS

Reviewed By: ZolotukhinM

Differential Revision: D30652220

Pulled By: huiguoo

fbshipit-source-id: cd75005cdfa42751318de7174b44e14a3a01634e
test/cpp/tensorexpr/test_kernel.cpp diff | blob | history
torch/csrc/jit/tensorexpr/kernel.cpp diff | blob | history
torch/csrc/jit/tensorexpr/kernel.h diff | blob | history
Domain: Machine Learning / ML Framework;