Add ATen and c10 to clang format config. (#9719)

This is a follow-up from #9715.

This PR fixes the mentioned PR by adding both `ATen` and `c10` includes
to the `.clang-format` file. They were previously mentioned in that
file's comments, but were not added to the rules.

Author: ysiraichi

.clang-format

@@ -102,6 +102,10 @@ IncludeBlocks:   Regroup
 IncludeCategories:
   - Regex:           '^<torch/.*>'
     Priority:        3
+  - Regex:           '^<ATen/.*>'
+    Priority:        3
+  - Regex:           '^<c10/.*>'
+    Priority:        3
   - Regex:           '^<.*\.h>'
     Priority:        1
   - Regex:           '^<.*'

test/cpp/test_replication.cpp

@@ -1,8 +1,9 @@
-#include <ATen/ATen.h>
 #include <gtest/gtest.h>
 
 #include <iostream>
 
+#include <ATen/ATen.h>
+
 #include "absl/synchronization/blocking_counter.h"
 #include "xla/hlo/builder/xla_builder.h"
 #include "xla/shape_util.h"

test/cpp/test_status_common.h

@@ -18,7 +18,6 @@
 #ifndef XLA_TEST_CPP_TEST_STATUS_COMMON_H_
 #define XLA_TEST_CPP_TEST_STATUS_COMMON_H_
 
-#include <c10/util/Exception.h>
 #include <gmock/gmock.h>
 #include <gtest/gtest.h>
 
@@ -27,6 +26,8 @@
 #include <stdexcept>
 #include <utility>
 
+#include <c10/util/Exception.h>
+
 #include "absl/status/status.h"
 #include "absl/status/statusor.h"
 

test/cpp/test_tensor.cpp

@@ -1,9 +1,9 @@
-#include <ATen/ATen.h>
 #include <gtest/gtest.h>
 
 #include <limits>
 #include <vector>
 
+#include <ATen/ATen.h>
 #include <torch/csrc/autograd/variable.h>
 
 #include "absl/base/nullability.h"

test/cpp/test_xla_sharding.cpp

@@ -1,10 +1,10 @@
-#include <ATen/ATen.h>
 #include <google/protobuf/repeated_field.h>
 #include <gtest/gtest.h>
 #include <stdlib.h>
 
 #include <iostream>
 
+#include <ATen/ATen.h>
 #include <torch/csrc/lazy/core/lazy_graph_executor.h>
 
 #include "xla/protobuf_util.h"

torch_xla/csrc/aten_fallback.cpp

@@ -1,13 +1,12 @@
 #include "torch_xla/csrc/aten_fallback.h"
 
+#include <unordered_map>
+#include <vector>
+
 #include <ATen/DLConvertor.h>
 #include <ATen/native/CPUFallback.h>
 #include <ATen/ops/_copy_from_and_resize.h>
 #include <ATen/ops/_to_cpu.h>
-
-#include <unordered_map>
-#include <vector>
-
 #include <torch/csrc/utils/device_lazy_init.h>
 
 #include "torch_xla/csrc/function_call_tracker.h"

torch_xla/csrc/aten_xla_bridge.cpp

@@ -1,11 +1,10 @@
 #include "torch_xla/csrc/aten_xla_bridge.h"
 
-#include <ATen/FunctionalTensorWrapper.h>
-
 #include <map>
 #include <string>
 #include <vector>
 
+#include <ATen/FunctionalTensorWrapper.h>
 #include <torch/csrc/lazy/core/tensor_util.h>
 
 #include "absl/log/absl_check.h"

torch_xla/csrc/aten_xla_bridge.h

@@ -1,12 +1,12 @@
 #ifndef XLA_TORCH_XLA_CSRC_ATEN_XLA_BRIDGE_H_
 #define XLA_TORCH_XLA_CSRC_ATEN_XLA_BRIDGE_H_
 
+#include <vector>
+
 #include <ATen/Device.h>
 #include <ATen/Functions.h>
 #include <ATen/Tensor.h>
 
-#include <vector>
-
 #include "absl/base/nullability.h"
 #include "absl/status/statusor.h"
 #include "absl/types/span.h"

torch_xla/csrc/aten_xla_type.cpp

@@ -1,3 +1,7 @@
+#include <iterator>
+#include <mutex>
+#include <optional>
+
 #include <ATen/ExpandUtils.h>
 #include <ATen/FunctionalTensorWrapper.h>
 #include <ATen/MetaFunctions.h>
@@ -10,11 +14,6 @@
 #include <ATen/ops/_embedding_bag_backward_native.h>
 #include <ATen/ops/expand_copy.h>
 #include <c10/core/Contiguity.h>
-
-#include <iterator>
-#include <mutex>
-#include <optional>
-
 #include <torch/csrc/lazy/core/helpers.h>
 #include <torch/csrc/lazy/core/shape_inference.h>
 #include <torch/csrc/lazy/core/tensor_util.h>

torch_xla/csrc/autocast_mode.cpp

@@ -4,7 +4,6 @@
 #include <ATen/autocast_mode.h>
 #include <c10/core/impl/LocalDispatchKeySet.h>
 #include <c10/util/intrusive_ptr.h>
-
 #include <torch/library.h>
 
 namespace at {