From 8136c39b5e22a307f5aa50b1c1ca6a4a11283eee Mon Sep 17 00:00:00 2001
From: Sebastian Messmer <messmer@fb.com>
Date: Thu, 10 Jan 2019 16:06:26 -0800
Subject: 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
---
 tools/build_variables.py | 1 +
 1 file changed, 1 insertion(+)

(limited to 'tools')

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",
 ]
-- 
cgit v1.2.3