projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 92cd5ab) | patch
[tensorexpr] Add 'pre_alloc' argument in python API of tensorexpr kernel (#64718)
authorHui Guo <huiguo@fb.com>
Fri, 10 Sep 2021 16:59:25 +0000 (09:59 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Fri, 10 Sep 2021 17:03:00 +0000 (10:03 -0700)
commitade4bf3e82ea369e77d511a796ba936ce281aa89
treee242069de54582f961050413117644cc75f01ecctree | snapshot
parent92cd5ab1cb2431d6d1c828d587b09db54077ed1fcommit | diff
[tensorexpr] Add 'pre_alloc' argument in python API of tensorexpr kernel (#64718)

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

Test Plan: Imported from OSS

Reviewed By: navahgar

Differential Revision: D30826582

Pulled By: huiguoo

fbshipit-source-id: 6c173c8964f2643039273cdc83e64fb02bb5f381
torch/csrc/jit/tensorexpr/tensorexpr_init.cpp diff | blob | history
Domain: Machine Learning / ML Framework;