pytorch/test/cpp
Cheng Chang 109bc1047e [NNC] Generate C++ code for Allocate and Free (#51070)
Summary:
This is the initial skeleton for C++ codegen, it includes generations for Allocate and Free.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/51070

Test Plan: New unit tests are added to `test_cpp_codegen.cpp`.

Reviewed By: ZolotukhinM

Differential Revision: D26061818

Pulled By: cheng-chang

fbshipit-source-id: b5256b2dcee6b2583ba73b6c9684994dbe7cdc1f
2021-02-01 13:06:51 -08:00
..
api fix unflatten_dense_tensor when there is empty tensor inside (#50321) 2021-01-23 12:14:34 -08:00
common
dist_autograd
jit Support complex number list in JIT (#51145) 2021-01-31 23:54:14 -08:00
rpc Support device map for distributed autograd while using TensorPipe. (#44859) 2021-01-27 13:01:44 -08:00
tensorexpr [NNC] Generate C++ code for Allocate and Free (#51070) 2021-02-01 13:06:51 -08:00
__init__.py