Code reformat

llvm · Jul 25, 2022 · bfe533f · bfe533f
1 parent 6d21b8d
commit bfe533f
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 6 deletions.
diff --git a/python/torch_mlir/csrc/base_lazy_backend/mlir_lowering_context.cpp b/python/torch_mlir/csrc/base_lazy_backend/mlir_lowering_context.cpp
@@ -52,11 +52,11 @@ TorchMlirLoweringContext::TorchMlirLoweringContext(
       function_(
           std::make_shared<torch::jit::GraphFunction>(name, graph_, nullptr)),
       mlir_context_(mlirContextCreate()) {
+  RegisterMlirDialects();
+
   for (auto node : post_order) {
     Lower(node);
   }
-
-  RegisterMlirDialects();
 }
 
 void TorchMlirLoweringContext::Lower(const Node* node) {

diff --git a/python/torch_mlir/csrc/base_lazy_backend/mlir_lowering_context.h b/python/torch_mlir/csrc/base_lazy_backend/mlir_lowering_context.h
@@ -137,7 +137,7 @@ class TORCH_API TorchMlirComputation : public torch::lazy::Computation {
 
   const std::string debug_string() const;
 
-  const std::string to_string() const;
+  const std::string to_string() const override;
 
 private:
   std::vector<std::string> parameter_names_;

diff --git a/python/torch_mlir/csrc/reference_lazy_backend/backend_impl.cpp b/python/torch_mlir/csrc/reference_lazy_backend/backend_impl.cpp
@@ -32,7 +32,9 @@ struct ReferenceLazyBackendDeviceType : public BackendDeviceType {
   ReferenceLazyBackendDeviceType(int8_t device_type)
       : device_type_(static_cast<c10::DeviceType>(device_type)) {}
 
-  std::string toString() const override { return c10::DeviceTypeName(device_type_); }
+  std::string toString() const override {
+    return c10::DeviceTypeName(device_type_);
+  }
 
   c10::DeviceType device_type_;
 };
@@ -127,11 +129,11 @@ class ReferenceLazyBackendImpl : public torch::lazy::TorchMlirBackendImpl {
   /**
    * Device Configuration
    * */
-  std::shared_ptr<torch::lazy::BackendDeviceType> GetDefaultDeviceType() const {
+  std::shared_ptr<torch::lazy::BackendDeviceType>
+  GetDefaultDeviceType() const override {
     return std::make_shared<BackendDeviceType>(default_device_type_);
   }
 
-
   void SetDefaultDeviceType(int8_t device_type) override {
     default_device_type_ = ReferenceLazyBackendDeviceType(device_type);
   }