diff options
| author | Sebastian Messmer <messmer@fb.com> | 2019-01-10 16:06:26 -0800 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-01-10 16:22:18 -0800 |
| commit | 8136c39b5e22a307f5aa50b1c1ca6a4a11283eee (patch) | |
| tree | e90fd0f6dce251c8ff02257182fd30c7f2f93ae9 /tools | |
| parent | 913785445eadcd8de1b55a5329e4424847565626 (diff) | |
| download | pytorch-8136c39b5e22a307f5aa50b1c1ca6a4a11283eee.tar.gz pytorch-8136c39b5e22a307f5aa50b1c1ca6a4a11283eee.tar.bz2 pytorch-8136c39b5e22a307f5aa50b1c1ca6a4a11283eee.zip | |
Enable calling caffe2 LayerNorm from PyTorch and JIT (#15243)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15243
Register it as a custom JIT op.
Reviewed By: dzhulgakov
Differential Revision: D13473791
fbshipit-source-id: 0f7e72e3efc85a75060a7597fadaf0a8bd289651
Diffstat (limited to 'tools')
| -rw-r--r-- | tools/build_variables.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tools/build_variables.py b/tools/build_variables.py index 53c53a641d..08a2218fae 100644 --- a/tools/build_variables.py +++ b/tools/build_variables.py @@ -97,6 +97,7 @@ torch_sources_no_python_default = [ "torch/csrc/jit/script/lexer.cpp", "torch/csrc/jit/script/module.cpp", "torch/csrc/jit/tracer.cpp", + "torch/csrc/jit/c10_ops/layer_norm.cpp", "torch/csrc/utils/tensor_flatten.cpp", "torch/csrc/utils/variadic.cpp", ] |