Formatting

src/include/rtg/operators.hpp

@@ -426,12 +426,13 @@ struct broadcast
     std::string name() const { return "broadcast"; }
     shape compute_shape(std::vector<shape> inputs) const
     {
-        auto t              = inputs.at(0).type();
-        auto result         = inputs.at(0);
-        auto input         = inputs.at(1);
-        
+        auto t      = inputs.at(0).type();
+        auto result = inputs.at(0);
+        auto input  = inputs.at(1);
+
         std::vector<size_t> bcast_strides(result.lens().size(), 0);
-        if(std::all_of(result.lens().cbegin(), result.lens().cend(), [&](auto x) { return x == 1; }))
+        if(std::all_of(
+               result.lens().cbegin(), result.lens().cend(), [&](auto x) { return x == 1; }))
         {
             if(axis != 0)
                 RTG_THROW("when broadcasting tensor of size 1, axis should be 0");
@@ -439,10 +440,10 @@ struct broadcast
         }
         else
         {
-            assert(result.lens().size()-axis >= input.lens().size());
-            if(!std::equal(input.lens().begin(), input.lens().end(), result.lens().begin()+axis))
+            assert(result.lens().size() - axis >= input.lens().size());
+            if(!std::equal(input.lens().begin(), input.lens().end(), result.lens().begin() + axis))
                 RTG_THROW("when broadcasting success sizes must match");
-            std::copy(input.strides().begin(), input.strides().end(), bcast_strides.begin()+axis);
+            std::copy(input.strides().begin(), input.strides().end(), bcast_strides.begin() + axis);
             return {t, result.lens(), std::move(bcast_strides)};
         }
     }

test/cpu_ops_test.cpp

@@ -91,10 +91,10 @@ void broadcast_test()
     p.compile(rtg::cpu::cpu_target{});
     auto result = p.eval({});
     auto output = result.get<int32_t>();
-    EXPECT(output(0,0) == -2);
-    EXPECT(output(0,1) == -2);
-    EXPECT(output(1,0) == -3);
-    EXPECT(output(1,1) == -3);
+    EXPECT(output(0, 0) == -2);
+    EXPECT(output(0, 1) == -2);
+    EXPECT(output(1, 0) == -3);
+    EXPECT(output(1, 1) == -3);
 }
 void add_broadcast_test()
 {