formatting

Author: fschlimb

include/imex/Dialect/NDArray/Extensions/AllExtensions.h

@@ -21,8 +21,8 @@ class DialectRegistry;
 namespace imex {
 namespace ndarray {
 
-/// Register all extensions of the NDArray dialect. This should generally only be
-/// used by tools, or other use cases that really do want *all* extensions of
+/// Register all extensions of the NDArray dialect. This should generally only
+/// be used by tools, or other use cases that really do want *all* extensions of
 /// the dialect. All other cases should prefer to instead register the specific
 /// extensions they intend to take advantage of.
 void registerAllExtensions(mlir::DialectRegistry &registry);

test/Dialect/NDArray/Transforms/AddGPURegions.mlir

@@ -46,7 +46,7 @@ func.func @test_region(%arg0: i64, %arg1: i64, %arg2: i64) -> i64 {
       // CHECK-NEXT: [[v9:%.*]] = tensor.empty() : tensor<i64, #ndarray.envs<#region.gpu_env<device = "XeGPU">>>
       // CHECK-NEXT: region.env_region_yield [[v9]] : tensor<i64, #ndarray.envs<#region.gpu_env<device = "XeGPU">>>
     // CHECK: [[v7:%.*]] = region.env_region #region.gpu_env<device = "XeGPU"> -> tensor<i64, #ndarray.envs<#region.gpu_env<device = "XeGPU">>> {
-      // CHECK-NEXT: [[vreduced:%.*]] = linalg.reduce { arith.addi {overflowFlags = #arith.overflow<none>} } ins([[v5]] : tensor<22xi64, #ndarray.envs<#region.gpu_env<device = "XeGPU">>>) outs([[v6]] : tensor<i64, #ndarray.envs<#region.gpu_env<device = "XeGPU">>>) dimensions = [0] 
+      // CHECK-NEXT: [[vreduced:%.*]] = linalg.reduce { arith.addi {overflowFlags = #arith.overflow<none>} } ins([[v5]] : tensor<22xi64, #ndarray.envs<#region.gpu_env<device = "XeGPU">>>) outs([[v6]] : tensor<i64, #ndarray.envs<#region.gpu_env<device = "XeGPU">>>) dimensions = [0]
       // CHECK-NEXT: region.env_region_yield [[vreduced]] : tensor<i64, #ndarray.envs<#region.gpu_env<device = "XeGPU">>>
     // CHECK: [[v8:%.*]] = region.env_region #region.gpu_env<device = "XeGPU"> -> i64 {
       // CHECK-NEXT: [[v9:%.*]] = builtin.unrealized_conversion_cast [[v7]] : tensor<i64, #ndarray.envs<#region.gpu_env<device = "XeGPU">>> to i64

tools/imex-opt/imex-opt.cpp

@@ -25,10 +25,10 @@
 #include <mlir/Support/FileUtilities.h>
 #include <mlir/Tools/mlir-opt/MlirOptMain.h>
 
+#include "imex/Dialect/NDArray/Extensions/AllExtensions.h"
 #include <imex/InitIMEXDialects.h>
 #include <imex/InitIMEXPasses.h>
 #include <imex/Transforms/Passes.h>
-#include "imex/Dialect/NDArray/Extensions/AllExtensions.h"
 
 int main(int argc, char **argv) {
   ::mlir::registerAllPasses();