code backup

src/include/migraphx/operators.hpp

@@ -44,6 +44,7 @@
 #include <migraphx/op/outline.hpp>
 #include <migraphx/op/pad.hpp>
 #include <migraphx/op/pooling.hpp>
+#include <migraphx/op/pow.hpp>
 #include <migraphx/op/reduce_sum.hpp>
 #include <migraphx/op/relu.hpp>
 #include <migraphx/op/reshape.hpp>

src/onnx/onnx.cpp

@@ -62,6 +62,7 @@ struct onnx_parser
         add_variadic_op("Sum", op::add{});
         add_variadic_op("Max", op::max{});
         add_variadic_op("Min", op::min{});
+        add_variadic_op("Pow", op::pow{});
 
         add_mem_op("ArgMax", &onnx_parser::parse_argmax);
         add_mem_op("ArgMin", &onnx_parser::parse_argmin);
@@ -189,10 +190,6 @@ struct onnx_parser
                        s1.begin() + offset,
                        out_lens.begin() + offset,
                        [](auto a, auto b) {
-                           if(a != b and a != 1 and b != 1)
-                           {
-                               MIGRAPHX_THROW("COMPUTE_BROADCASTED_LEN: input shapes mismatch!");
-                           }
                            return std::max(a, b);
                        });
 

src/targets/gpu/CMakeLists.txt

@@ -38,6 +38,7 @@ add_library(migraphx_device
     device/sub.cpp
     device/clip.cpp
     device/reduce_sum.cpp
+    device/pow.cpp
 )
 set_target_properties(migraphx_device PROPERTIES EXPORT_NAME device)
 rocm_clang_tidy_check(migraphx_device)

src/targets/gpu/lowering.cpp

@@ -50,6 +50,7 @@
 #include <migraphx/gpu/convert.hpp>
 #include <migraphx/gpu/clip.hpp>
 #include <migraphx/gpu/reduce_sum.hpp>
+#include <migraphx/gpu/pow.hpp>
 #include <utility>
 #include <functional>
 #include <algorithm>
@@ -98,6 +99,7 @@ struct miopen_apply
         add_generic_op<hip_mul>("mul");
         add_generic_op<hip_max>("max");
         add_generic_op<hip_min>("min");
+        add_generic_op<hip_pow>("pow");
 
         add_extend_op<miopen_gemm, op::dot>("dot");
         add_extend_op<miopen_contiguous, op::contiguous>("contiguous");