[PyTorch] Remove unnecessary dispatcher.h include in torch/library.h

swolchok · swolchok · commit e533528c9a2d · 2021-01-26T16:38:56.000-08:00
It's unused. Differential Revision: [D25859010](https://our.internmc.facebook.com/intern/diff/D25859010/) **NOTE FOR REVIEWERS**: This PR has internal Facebook specific changes or comments, please review them on [Phabricator](https://our.internmc.facebook.com/intern/diff/D25859010/)! [ghstack-poisoned]
diff --git a/aten/src/ATen/BatchedFallback.cpp b/aten/src/ATen/BatchedFallback.cpp
@@ -1,6 +1,7 @@
 #include <ATen/BatchedFallback.h>
 #include <ATen/MatrixRef.h>
 #include <ATen/VmapTransforms.h>
+#include <ATen/core/dispatch/Dispatcher.h>
 #include <c10/util/llvmMathExtras.h>
 
 namespace at {
diff --git a/aten/src/ATen/core/dispatch/backend_fallback_test.cpp b/aten/src/ATen/core/dispatch/backend_fallback_test.cpp
@@ -3,6 +3,7 @@
 #include <ATen/ATen.h>
 #include <ATen/NativeFunctions.h>
 #include <ATen/Functions.h>
+#include <ATen/core/dispatch/Dispatcher.h>
 #include <ATen/core/op_registration/op_registration.h>
 #include <torch/library.h>
 
diff --git a/aten/src/ATen/core/library.cpp b/aten/src/ATen/core/library.cpp
@@ -1,5 +1,7 @@
 #include <torch/library.h>
 
+#include <ATen/core/dispatch/Dispatcher.h>
+
 namespace torch {
 
 namespace {
diff --git a/aten/src/ATen/templates/RegisterBackendSelect.cpp b/aten/src/ATen/templates/RegisterBackendSelect.cpp
@@ -1,11 +1,11 @@
 // We register ops with a higher priority dispatch key (BackendSelect) than the usual backend-specific keys (e.g. CPU)
 // which makes calls to the factory functions dispatch to here.
 // We then 'manually' compute a lower-priority to re-dispatch to (e.g. CPU) to get to the eventually correct backend.
-
 // ${generated_comment}
 
 #include <ATen/ATen.h>
 #include <ATen/Dispatch.h>
+#include <ATen/core/dispatch/Dispatcher.h>
 #include <torch/library.h>
 #include <c10/core/TensorOptions.h>
 
diff --git a/tools/codegen/gen.py b/tools/codegen/gen.py
@@ -1127,7 +1127,7 @@ def write_with_template(self, filename: str, template_fn: str,
             if isinstance(env, dict):
                 # TODO: Update the comment reference to the correct location
                 if 'generated_comment' not in env:
-                    comment = "@" + "generated by aten/src/ATen/gen.py"
+                    comment = "@" + "generated by tools/codegen/gen.py"
                     comment += " from {}".format(os.path.basename(template_fn))
                     env['generated_comment'] = comment
                 template = _read_template(os.path.join(self.template_dir, template_fn))
diff --git a/torch/csrc/utils/python_dispatch.cpp b/torch/csrc/utils/python_dispatch.cpp
@@ -3,6 +3,7 @@
 
 #include <torch/library.h>
 #include <ATen/ATen.h>
+#include <ATen/core/dispatch/Dispatcher.h>
 
 #include <pybind11/operators.h>
 
diff --git a/torch/library.h b/torch/library.h
@@ -59,7 +59,6 @@
 /// ```
 
 #include <c10/core/DispatchKey.h>
-#include <ATen/core/dispatch/Dispatcher.h>
 #include <ATen/core/op_registration/op_whitelist.h>
 #include <ATen/core/op_registration/infer_schema.h>
 #if defined(EXPOSE_C2_OPS) || !defined(CAFFE2_IS_XPLAT_BUILD)
