Commit 0e237605 authored by turneram's avatar turneram
Browse files

Formatting

parent 985fb0dd
...@@ -67,8 +67,8 @@ struct CKBinaryElementwise ...@@ -67,8 +67,8 @@ struct CKBinaryElementwise
template <class L, class S> template <class L, class S>
constexpr auto MakeDescriptor_M(const L& lengths, const S& strides) constexpr auto MakeDescriptor_M(const L& lengths, const S& strides)
{ {
auto tupleOfShape = generate_tuple([&](auto I) { return static_cast<ck::index_t>(lengths[I]); }, auto tupleOfShape = generate_tuple(
ck::Number<ndim>{}); [&](auto I) { return static_cast<ck::index_t>(lengths[I]); }, ck::Number<ndim>{});
auto tupleOfStride = generate_tuple( auto tupleOfStride = generate_tuple(
[&](auto I) { return static_cast<ck::index_t>(strides[I]); }, ck::Number<ndim>{}); [&](auto I) { return static_cast<ck::index_t>(strides[I]); }, ck::Number<ndim>{});
const auto desc = make_naive_tensor_descriptor(tupleOfShape, tupleOfStride); const auto desc = make_naive_tensor_descriptor(tupleOfShape, tupleOfStride);
......
...@@ -37,7 +37,8 @@ struct ck_elementwise_half : verify_program<ck_elementwise_half> ...@@ -37,7 +37,8 @@ struct ck_elementwise_half : verify_program<ck_elementwise_half>
migraphx::shape m2_shape{migraphx::shape::half_type, {3072}}; migraphx::shape m2_shape{migraphx::shape::half_type, {3072}};
auto l1 = mm->add_parameter("1", m1_shape); auto l1 = mm->add_parameter("1", m1_shape);
auto l2 = mm->add_parameter("2", m2_shape); auto l2 = mm->add_parameter("2", m2_shape);
l2 = mm->add_instruction(migraphx::make_op("multibroadcast", {{"out_lens", {2, 384, 3072}}}), l2); l2 = mm->add_instruction(
migraphx::make_op("multibroadcast", {{"out_lens", {2, 384, 3072}}}), l2);
mm->add_instruction(migraphx::make_op("ck_elementwise"), l1, l2); mm->add_instruction(migraphx::make_op("ck_elementwise"), l1, l2);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment