pytorch/test/cpp
Jiakai Liu 3c4d57c18b [pytorch][nnc] update external functions for mobile build (#56850)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/56850

This is part of the changes to enable NNC AOT compilation for mobile.
The generated kernels need to call these external functions thus change the declarations to use C linkage when building the mobile runtime.

Added nnc_aten_addmm external function.

ghstack-source-id: 127877411

Test Plan:
- build & CI;
- tested mobile build with stacked PRs;

Reviewed By: ZolotukhinM

Differential Revision: D27897154

fbshipit-source-id: 61d5499d7781a83bd2657859659fd1b5043d6b04
2021-04-30 19:07:19 -07:00
..
api Make PyTorch code-base clang-tidy compliant (#56892) 2021-04-28 14:10:25 -07:00
common
dist_autograd Forbid trailing whitespace (#53406) 2021-03-05 17:22:55 -08:00
jit [AutoAccept][Codemod][FBSourceClangFormatLinter] Daily arc lint --take CLANGFORMAT 2021-04-29 04:12:25 -07:00
lite_interpreter_runtime Revert D27958477: [PyTorch][Edge] Add v4 and v5 models and remove unused model 2021-04-23 14:42:01 -07:00
rpc [PyTorch][jit] Don't allow create() on singleton types (#56807) 2021-04-30 10:28:50 -07:00
tensorexpr [pytorch][nnc] update external functions for mobile build (#56850) 2021-04-30 19:07:19 -07:00
__init__.py