Proto data dependent symbolics - #5077
Draft
CharlieL7 wants to merge 52 commits into
Draft
Conversation
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
![github-actions[bot]](https://avatars.githubusercontent.com/in/15368?s=60&v=4){kind=small}
Comment on lines
+73
to
+74
| auto type = inputs.at(0).type(); | ||
| auto k_val = std::get<int64_t>(k); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| auto type = inputs.at(0).type(); | |
| auto k_val = std::get<int64_t>(k); | |
| auto type = inputs.at(0).type(); | |
| auto k_val = std::get<int64_t>(k); |
| const shape k_shape{shape::int64_type, {1}}; | ||
| auto k_lit = info.add_literal(literal{k_shape, {k}}); | ||
| auto topk_ret = info.add_instruction( | ||
| make_op("topk", {{"k", k}, {"axis", axis}, {"largest", largest}}), args.at(0), k_lit); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| make_op("topk", {{"k", k}, {"axis", axis}, {"largest", largest}}), args.at(0), k_lit); | |
| make_op("topk", {{"k", k}, {"axis", axis}, {"largest", largest}}), | |
| args.at(0), | |
| k_lit); |
Comment on lines
+71
to
+72
| auto ret_val = info.add_instruction(make_op("get_tuple_elem", {{"index", 0}}), topk_ret); | ||
| auto ret_ind = info.add_instruction(make_op("get_tuple_elem", {{"index", 1}}), topk_ret); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| auto ret_val = info.add_instruction(make_op("get_tuple_elem", {{"index", 0}}), topk_ret); | |
| auto ret_ind = info.add_instruction(make_op("get_tuple_elem", {{"index", 1}}), topk_ret); | |
| auto ret_val = | |
| info.add_instruction(make_op("get_tuple_elem", {{"index", 0}}), topk_ret); | |
| auto ret_ind = | |
| info.add_instruction(make_op("get_tuple_elem", {{"index", 1}}), topk_ret); |
| // Constant `k`: use its value for the attribute; topk output is already the exact size. | ||
| int64_t k = arg_k.at<int>(); | ||
| auto topk_ret = info.add_instruction( | ||
| make_op("topk", {{"k", k}, {"axis", axis}, {"largest", largest}}), args.at(0), args.at(1)); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| make_op("topk", {{"k", k}, {"axis", axis}, {"largest", largest}}), args.at(0), args.at(1)); | |
| make_op("topk", {{"k", k}, {"axis", axis}, {"largest", largest}}), | |
| args.at(0), | |
| args.at(1)); |
Comment on lines
+85
to
+86
| auto ret_val = info.add_instruction(make_op("get_tuple_elem", {{"index", 0}}), topk_ret); | ||
| auto ret_ind = info.add_instruction(make_op("get_tuple_elem", {{"index", 1}}), topk_ret); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| auto ret_val = info.add_instruction(make_op("get_tuple_elem", {{"index", 0}}), topk_ret); | |
| auto ret_ind = info.add_instruction(make_op("get_tuple_elem", {{"index", 1}}), topk_ret); | |
| auto ret_val = | |
| info.add_instruction(make_op("get_tuple_elem", {{"index", 0}}), topk_ret); | |
| auto ret_ind = | |
| info.add_instruction(make_op("get_tuple_elem", {{"index", 1}}), topk_ret); |
Comment on lines
+199
to
+200
| auto kk = mm->add_literal( | ||
| migraphx::literal{migraphx::shape{migraphx::shape::int64_type, {1}}, {5}}); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| auto kk = mm->add_literal( | |
| migraphx::literal{migraphx::shape{migraphx::shape::int64_type, {1}}, {5}}); | |
| auto kk = | |
| mm->add_literal(migraphx::literal{migraphx::shape{migraphx::shape::int64_type, {1}}, {5}}); |
Comment on lines
202
to
+203
| auto r = mm->add_instruction(migraphx::make_op("topk", {{"axis", 1}, {"k", 5}, {"largest", 0}}), | ||
| data); | ||
| data, kk); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| auto r = mm->add_instruction(migraphx::make_op("topk", {{"axis", 1}, {"k", 5}, {"largest", 0}}), | |
| data); | |
| data, kk); | |
| auto r = mm->add_instruction( | |
| migraphx::make_op("topk", {{"axis", 1}, {"k", 5}, {"largest", 0}}), data, kk); |
Comment on lines
+98
to
+99
| auto r2 = m2.add_instruction( | ||
| migraphx::make_op("topk", {{"k", 8}, {"axis", 0}}), valuer, k, idxr); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| auto r2 = m2.add_instruction( | |
| migraphx::make_op("topk", {{"k", 8}, {"axis", 0}}), valuer, k, idxr); | |
| auto r2 = | |
| m2.add_instruction(migraphx::make_op("topk", {{"k", 8}, {"axis", 0}}), valuer, k, idxr); |
Comment on lines
+140
to
+141
| auto r2 = m2.add_instruction( | ||
| migraphx::make_op("topk", {{"k", 8}, {"axis", 1}}), valuer, k, idxr); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| auto r2 = m2.add_instruction( | |
| migraphx::make_op("topk", {{"k", 8}, {"axis", 1}}), valuer, k, idxr); | |
| auto r2 = | |
| m2.add_instruction(migraphx::make_op("topk", {{"k", 8}, {"axis", 1}}), valuer, k, idxr); |
Comment on lines
+182
to
+183
| auto r2 = m2.add_instruction( | ||
| migraphx::make_op("topk", {{"k", 8}, {"axis", 0}}), valuer, k, idxr); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| auto r2 = m2.add_instruction( | |
| migraphx::make_op("topk", {{"k", 8}, {"axis", 0}}), valuer, k, idxr); | |
| auto r2 = | |
| m2.add_instruction(migraphx::make_op("topk", {{"k", 8}, {"axis", 0}}), valuer, k, idxr); |
| } | ||
|
|
||
| template <class E, MIGRAPHX_REQUIRES(is_bit_flag<E>{})> | ||
| constexpr E& operator|=(E& lhs, E rhs) |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| constexpr E& operator|=(E& lhs, E rhs) | |
| constexpr E& operator|=(E & lhs, E rhs) |
| } | ||
|
|
||
| template <class E, MIGRAPHX_REQUIRES(is_bit_flag<E>{})> | ||
| constexpr E& operator&=(E& lhs, E rhs) |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| constexpr E& operator&=(E& lhs, E rhs) | |
| constexpr E& operator&=(E & lhs, E rhs) |
| } | ||
|
|
||
| template <class E, MIGRAPHX_REQUIRES(is_bit_flag<E>{})> | ||
| constexpr E& operator^=(E& lhs, E rhs) |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| constexpr E& operator^=(E& lhs, E rhs) | |
| constexpr E& operator^=(E & lhs, E rhs) |
|
|
||
| op::slice::slice_mode get_slice_mode(slice_input_flags slice_flags) | ||
| { | ||
| switch (slice_flags) |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| switch (slice_flags) | |
| switch(slice_flags) |
Comment on lines
+81
to
+98
| case slice_input_flags::none: | ||
| return op::slice::slice_mode::one_input; | ||
| case slice_input_flags::starts_input: | ||
| return op::slice::slice_mode::starts_input; | ||
| case slice_input_flags::ends_input: | ||
| return op::slice::slice_mode::ends_input; | ||
| case slice_input_flags::axes_input: | ||
| return op::slice::slice_mode::axes_input; | ||
| case (slice_input_flags::starts_input | slice_input_flags::ends_input): | ||
| return op::slice::slice_mode::starts_ends_input; | ||
| case (slice_input_flags::starts_input | slice_input_flags::axes_input): | ||
| return op::slice::slice_mode::starts_axes_input; | ||
| case (slice_input_flags::ends_input | slice_input_flags::axes_input): | ||
| return op::slice::slice_mode::ends_axes_input; | ||
| case (slice_input_flags::starts_input | slice_input_flags::ends_input | slice_input_flags::axes_input): | ||
| return op::slice::slice_mode::starts_ends_axes_input; | ||
| default: | ||
| MIGRAPHX_THROW("PARSE_SLICE: invalid slice_mode"); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| case slice_input_flags::none: | |
| return op::slice::slice_mode::one_input; | |
| case slice_input_flags::starts_input: | |
| return op::slice::slice_mode::starts_input; | |
| case slice_input_flags::ends_input: | |
| return op::slice::slice_mode::ends_input; | |
| case slice_input_flags::axes_input: | |
| return op::slice::slice_mode::axes_input; | |
| case (slice_input_flags::starts_input | slice_input_flags::ends_input): | |
| return op::slice::slice_mode::starts_ends_input; | |
| case (slice_input_flags::starts_input | slice_input_flags::axes_input): | |
| return op::slice::slice_mode::starts_axes_input; | |
| case (slice_input_flags::ends_input | slice_input_flags::axes_input): | |
| return op::slice::slice_mode::ends_axes_input; | |
| case (slice_input_flags::starts_input | slice_input_flags::ends_input | slice_input_flags::axes_input): | |
| return op::slice::slice_mode::starts_ends_axes_input; | |
| default: | |
| MIGRAPHX_THROW("PARSE_SLICE: invalid slice_mode"); | |
| case slice_input_flags::none: return op::slice::slice_mode::one_input; | |
| case slice_input_flags::starts_input: return op::slice::slice_mode::starts_input; | |
| case slice_input_flags::ends_input: return op::slice::slice_mode::ends_input; | |
| case slice_input_flags::axes_input: return op::slice::slice_mode::axes_input; | |
| case(slice_input_flags::starts_input | slice_input_flags::ends_input): | |
| return op::slice::slice_mode::starts_ends_input; | |
| case(slice_input_flags::starts_input | slice_input_flags::axes_input): | |
| return op::slice::slice_mode::starts_axes_input; | |
| case(slice_input_flags::ends_input | slice_input_flags::axes_input): | |
| return op::slice::slice_mode::ends_axes_input; | |
| case(slice_input_flags::starts_input | slice_input_flags::ends_input | | |
| slice_input_flags::axes_input): | |
| return op::slice::slice_mode::starts_ends_axes_input; | |
| default: MIGRAPHX_THROW("PARSE_SLICE: invalid slice_mode"); |
| op::slice create_slice_operator() | ||
| { | ||
| op::slice slice_op; | ||
| slice_op.axes = axes; |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| slice_op.axes = axes; | |
| slice_op.axes = axes; |
Comment on lines
+107
to
+108
| slice_op.ends = ends; | ||
| slice_op.mode = get_slice_mode(flags); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| slice_op.ends = ends; | |
| slice_op.mode = get_slice_mode(flags); | |
| slice_op.ends = ends; | |
| slice_op.mode = get_slice_mode(flags); |
Contributor
There was a problem hiding this comment.
Remaining comments which cannot be posted as a review comment to avoid GitHub Rate Limit
format.py
[format.py] reported by reviewdog 🐶
AMDMIGraphX/test/op_shape_test.cpp
Lines 5226 to 5230 in 9a224a2
[format.py] reported by reviewdog 🐶
AMDMIGraphX/test/ref/slice.cpp
Lines 90 to 94 in 9a224a2
[format.py] reported by reviewdog 🐶
AMDMIGraphX/test/ref/slice.cpp
Lines 120 to 124 in 9a224a2
[format.py] reported by reviewdog 🐶
AMDMIGraphX/test/ref/slice.cpp
Lines 262 to 266 in 9a224a2
[format.py] reported by reviewdog 🐶
AMDMIGraphX/test/ref/slice.cpp
Lines 327 to 331 in 9a224a2
[format.py] reported by reviewdog 🐶
AMDMIGraphX/test/ref/slice.cpp
Lines 359 to 363 in 9a224a2
[format.py] reported by reviewdog 🐶
AMDMIGraphX/test/simplify_dyn_ops_test.cpp
Lines 537 to 542 in 9a224a2
| if(inputs.size() == 1) | ||
| { | ||
| if(any_sym(starts) or any_sym(ends)) | ||
| MIGRAPHX_THROW("SLICE: Invalid attributes: symbolic in attribute for 1 input slice"); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| MIGRAPHX_THROW("SLICE: Invalid attributes: symbolic in attribute for 1 input slice"); | |
| MIGRAPHX_THROW( | |
| "SLICE: Invalid attributes: symbolic in attribute for 1 input slice"); |
| if(inputs.size() == 2) | ||
| { | ||
| if(set_attributes == ends_axes) | ||
| std::vector<slice_mode> two_input_modes_not_axes = {slice_mode::starts_input, slice_mode::ends_input}; |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| std::vector<slice_mode> two_input_modes_not_axes = {slice_mode::starts_input, slice_mode::ends_input}; | |
| std::vector<slice_mode> two_input_modes_not_axes = {slice_mode::starts_input, | |
| slice_mode::ends_input}; |
| dds.at(axis) = {0, dds.at(axis).get_interval().max}; | ||
| }); | ||
| if(inputs[1].lens()[0] != axes.size()) | ||
| MIGRAPHX_THROW("SLICE: input length (" + migraphx::to_string(inputs[1].lens()[0]) + ") does not match attribute length (" + migraphx::to_string(axes.size()) + ")"); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| MIGRAPHX_THROW("SLICE: input length (" + migraphx::to_string(inputs[1].lens()[0]) + ") does not match attribute length (" + migraphx::to_string(axes.size()) + ")"); | |
| MIGRAPHX_THROW("SLICE: input length (" + | |
| migraphx::to_string(inputs[1].lens()[0]) + | |
| ") does not match attribute length (" + | |
| migraphx::to_string(axes.size()) + ")"); |
| return shape::dynamic_dimension{0, dd.get_interval().max}; | ||
| }); | ||
| if(inputs[1].lens()[0] != starts.size()) | ||
| MIGRAPHX_THROW("SLICE: input length (" + migraphx::to_string(inputs[1].lens()[0]) + ") does not match attribute length (" + migraphx::to_string(starts.size()) + ")"); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| MIGRAPHX_THROW("SLICE: input length (" + migraphx::to_string(inputs[1].lens()[0]) + ") does not match attribute length (" + migraphx::to_string(starts.size()) + ")"); | |
| MIGRAPHX_THROW("SLICE: input length (" + | |
| migraphx::to_string(inputs[1].lens()[0]) + | |
| ") does not match attribute length (" + | |
| migraphx::to_string(starts.size()) + ")"); |
| dds.at(axis) = {0, dds.at(axis).get_interval().max}; | ||
| }); | ||
| if(inputs[1].lens()[0] != axes.size()) | ||
| MIGRAPHX_THROW("SLICE: input length (" + migraphx::to_string(inputs[1].lens()[0]) + ") does not match attribute length (" + migraphx::to_string(axes.size()) + ")"); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| MIGRAPHX_THROW("SLICE: input length (" + migraphx::to_string(inputs[1].lens()[0]) + ") does not match attribute length (" + migraphx::to_string(axes.size()) + ")"); | |
| MIGRAPHX_THROW("SLICE: input length (" + | |
| migraphx::to_string(inputs[1].lens()[0]) + | |
| ") does not match attribute length (" + | |
| migraphx::to_string(axes.size()) + ")"); |
| auto ends = m1.add_parameter("ends", idx_s); | ||
| auto sl = | ||
| m1.add_instruction(migraphx::make_op("slice", {{"axes", {2}}}), data, starts, ends); | ||
| auto sl = m1.add_instruction( |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| auto sl = m1.add_instruction( | |
| auto sl = m1.add_instruction( |
| auto ends = mm->add_parameter("ends", migraphx::shape{migraphx::shape::int32_type, {2}}); | ||
| auto ret = | ||
| mm->add_instruction(migraphx::make_op("slice", {{"axes", {0, 1}}}), data, starts, ends); | ||
| auto ret = mm->add_instruction( |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| auto ret = mm->add_instruction( | |
| auto ret = mm->add_instruction( |
| auto ends = mm->add_parameter("ends", migraphx::shape{migraphx::shape::int32_type, {2}}); | ||
| auto axes = mm->add_parameter("axes", migraphx::shape{migraphx::shape::int32_type, {2}}); | ||
| auto ret = mm->add_instruction(migraphx::make_op("slice"), data, starts, ends, axes); | ||
| auto ret = mm->add_instruction( |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| auto ret = mm->add_instruction( | |
| auto ret = mm->add_instruction( |
Comment on lines
+41
to
+54
| val = mm->add_instruction(migraphx::make_op("slice", | ||
| {{"axes", {1}}, | ||
| {"starts", {0}}, | ||
| {"ends", migraphx::value::array{migraphx::to_value(k_var)}}, | ||
| {"mode", "ends_input"}}), | ||
| val, | ||
| k); | ||
| ind = mm->add_instruction(migraphx::make_op("slice", | ||
| {{"axes", {1}}, | ||
| {"starts", {0}}, | ||
| {"ends", migraphx::value::array{migraphx::to_value(k_var)}}, | ||
| {"mode", "ends_input"}}), | ||
| ind, | ||
| k); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| val = mm->add_instruction(migraphx::make_op("slice", | |
| {{"axes", {1}}, | |
| {"starts", {0}}, | |
| {"ends", migraphx::value::array{migraphx::to_value(k_var)}}, | |
| {"mode", "ends_input"}}), | |
| val, | |
| k); | |
| ind = mm->add_instruction(migraphx::make_op("slice", | |
| {{"axes", {1}}, | |
| {"starts", {0}}, | |
| {"ends", migraphx::value::array{migraphx::to_value(k_var)}}, | |
| {"mode", "ends_input"}}), | |
| ind, | |
| k); | |
| val = mm->add_instruction( | |
| migraphx::make_op("slice", | |
| {{"axes", {1}}, | |
| {"starts", {0}}, | |
| {"ends", migraphx::value::array{migraphx::to_value(k_var)}}, | |
| {"mode", "ends_input"}}), | |
| val, | |
| k); | |
| ind = mm->add_instruction( | |
| migraphx::make_op("slice", | |
| {{"axes", {1}}, | |
| {"starts", {0}}, | |
| {"ends", migraphx::value::array{migraphx::to_value(k_var)}}, | |
| {"mode", "ends_input"}}), | |
| ind, | |
| k); |
Comment on lines
+75
to
+88
| val = mm->add_instruction(migraphx::make_op("slice", | ||
| {{"axes", {1}}, | ||
| {"starts", {0}}, | ||
| {"ends", migraphx::value::array{migraphx::to_value(k_var)}}, | ||
| {"mode", "ends_input"}}), | ||
| val, | ||
| k); | ||
| ind = mm->add_instruction(migraphx::make_op("slice", | ||
| {{"axes", {1}}, | ||
| {"starts", {0}}, | ||
| {"ends", migraphx::value::array{migraphx::to_value(k_var)}}, | ||
| {"mode", "ends_input"}}), | ||
| ind, | ||
| k); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| val = mm->add_instruction(migraphx::make_op("slice", | |
| {{"axes", {1}}, | |
| {"starts", {0}}, | |
| {"ends", migraphx::value::array{migraphx::to_value(k_var)}}, | |
| {"mode", "ends_input"}}), | |
| val, | |
| k); | |
| ind = mm->add_instruction(migraphx::make_op("slice", | |
| {{"axes", {1}}, | |
| {"starts", {0}}, | |
| {"ends", migraphx::value::array{migraphx::to_value(k_var)}}, | |
| {"mode", "ends_input"}}), | |
| ind, | |
| k); | |
| val = mm->add_instruction( | |
| migraphx::make_op("slice", | |
| {{"axes", {1}}, | |
| {"starts", {0}}, | |
| {"ends", migraphx::value::array{migraphx::to_value(k_var)}}, | |
| {"mode", "ends_input"}}), | |
| val, | |
| k); | |
| ind = mm->add_instruction( | |
| migraphx::make_op("slice", | |
| {{"axes", {1}}, | |
| {"starts", {0}}, | |
| {"ends", migraphx::value::array{migraphx::to_value(k_var)}}, | |
| {"mode", "ends_input"}}), | |
| ind, | |
| k); |
Adds a pytest bridge to support running of CTest unit tests with pytest.
Fix the reference nonzero operator so it accepts and correctly reads non-standard input layouts such as transposed and broadcasted tensors.
Regressions detected 🔴 |
|
Macro builders were recently added to better manage composed ops when parsing from outside libraries. This extends the torch kit with a number of missing ops required to fully migrate torch_migraphx to use this builder framework.
Adds gfx115 to hipblaslt_supported_impl() so gfx1150/1151/1152/1153 (Strix Halo/Point, RDNA3.5) use the hipBLASLt GEMM path.
…into sym_resolve_op
…nd update tests (#5090) Rewrites the flash decoding kernel 2 recombination step in find_flash_decoding to use the exp-normalize form: `O = sum(O' * exp(LSE - max)) / sum(exp(LSE - max))` instead of normalizing weights first, then scaling and summing partial outputs. The result is mathematically equivalent but produces IR that fuses more cleanly downstream (e.g. with rewrite_broadcast in a follow-up PR).
…to_data_dependent_symbolics
Comment on lines
+56
to
+64
| return info.add_instruction( | ||
| make_op( | ||
| "slice", | ||
| {{"axes", {0}}, | ||
| {"starts", {0}}, | ||
| {"ends", value::array{to_value(num_selected_var)}}, | ||
| {"mode", "ends_input"}}), | ||
| indices, | ||
| num_selected); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| return info.add_instruction( | |
| make_op( | |
| "slice", | |
| {{"axes", {0}}, | |
| {"starts", {0}}, | |
| {"ends", value::array{to_value(num_selected_var)}}, | |
| {"mode", "ends_input"}}), | |
| indices, | |
| num_selected); | |
| return info.add_instruction(make_op("slice", | |
| {{"axes", {0}}, | |
| {"starts", {0}}, | |
| {"ends", value::array{to_value(num_selected_var)}}, | |
| {"mode", "ends_input"}}), | |
| indices, | |
| num_selected); |
Comment on lines
+5376
to
+5380
| throws_shape( | ||
| migraphx::make_op("slice", | ||
| {{"ends", {2, 3, 4}}, {"axes", {0, 1, 2}}, {"mode", "starts_input"}}), | ||
| input, | ||
| starts); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| throws_shape( | |
| migraphx::make_op("slice", | |
| {{"ends", {2, 3, 4}}, {"axes", {0, 1, 2}}, {"mode", "starts_input"}}), | |
| input, | |
| starts); | |
| throws_shape(migraphx::make_op( | |
| "slice", {{"ends", {2, 3, 4}}, {"axes", {0, 1, 2}}, {"mode", "starts_input"}}), | |
| input, | |
| starts); |
Comment on lines
+5387
to
+5391
| throws_shape( | ||
| migraphx::make_op("slice", | ||
| {{"starts", {0, 1, 2}}, {"axes", {0, 1, 2}}, {"mode", "ends_input"}}), | ||
| input, | ||
| ends); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| throws_shape( | |
| migraphx::make_op("slice", | |
| {{"starts", {0, 1, 2}}, {"axes", {0, 1, 2}}, {"mode", "ends_input"}}), | |
| input, | |
| ends); | |
| throws_shape(migraphx::make_op( | |
| "slice", {{"starts", {0, 1, 2}}, {"axes", {0, 1, 2}}, {"mode", "ends_input"}}), | |
| input, | |
| ends); |
Comment on lines
+5398
to
+5402
| throws_shape( | ||
| migraphx::make_op("slice", | ||
| {{"starts", {0, 1, 2}}, {"ends", {3, 4, 4}}, {"mode", "axes_input"}}), | ||
| input, | ||
| axes); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| throws_shape( | |
| migraphx::make_op("slice", | |
| {{"starts", {0, 1, 2}}, {"ends", {3, 4, 4}}, {"mode", "axes_input"}}), | |
| input, | |
| axes); | |
| throws_shape(migraphx::make_op( | |
| "slice", {{"starts", {0, 1, 2}}, {"ends", {3, 4, 4}}, {"mode", "axes_input"}}), | |
| input, | |
| axes); |
Comment on lines
+5410
to
+5414
| throws_shape( | ||
| migraphx::make_op("slice", {{"axes", {0, 1, 2}}, {"mode", "starts_ends_input"}}), | ||
| input, | ||
| starts, | ||
| ends); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| throws_shape( | |
| migraphx::make_op("slice", {{"axes", {0, 1, 2}}, {"mode", "starts_ends_input"}}), | |
| input, | |
| starts, | |
| ends); | |
| throws_shape(migraphx::make_op("slice", {{"axes", {0, 1, 2}}, {"mode", "starts_ends_input"}}), | |
| input, | |
| starts, | |
| ends); |
Comment on lines
+122
to
+126
| mm->add_instruction( | ||
| migraphx::make_op("slice", {{"axes", {2}}, {"mode", "starts_ends_input"}}), | ||
| l0, | ||
| starts, | ||
| ends); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| mm->add_instruction( | |
| migraphx::make_op("slice", {{"axes", {2}}, {"mode", "starts_ends_input"}}), | |
| l0, | |
| starts, | |
| ends); | |
| mm->add_instruction(migraphx::make_op("slice", {{"axes", {2}}, {"mode", "starts_ends_input"}}), | |
| l0, | |
| starts, | |
| ends); |
Comment on lines
+264
to
+268
| mm->add_instruction( | ||
| migraphx::make_op("slice", {{"axes", {2}}, {"mode", "starts_ends_input"}}), | ||
| input, | ||
| starts, | ||
| ends); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| mm->add_instruction( | |
| migraphx::make_op("slice", {{"axes", {2}}, {"mode", "starts_ends_input"}}), | |
| input, | |
| starts, | |
| ends); | |
| mm->add_instruction(migraphx::make_op("slice", {{"axes", {2}}, {"mode", "starts_ends_input"}}), | |
| input, | |
| starts, | |
| ends); |
Comment on lines
+329
to
+333
| mm->add_instruction( | ||
| migraphx::make_op("slice", {{"starts", {-4}}, {"mode", "ends_axes_input"}}), | ||
| input, | ||
| ends, | ||
| axes); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| mm->add_instruction( | |
| migraphx::make_op("slice", {{"starts", {-4}}, {"mode", "ends_axes_input"}}), | |
| input, | |
| ends, | |
| axes); | |
| mm->add_instruction(migraphx::make_op("slice", {{"starts", {-4}}, {"mode", "ends_axes_input"}}), | |
| input, | |
| ends, | |
| axes); |
Comment on lines
+361
to
+365
| mm->add_instruction( | ||
| migraphx::make_op("slice", {{"axes", {2}}, {"mode", "starts_ends_input"}}), | ||
| input, | ||
| starts, | ||
| ends); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| mm->add_instruction( | |
| migraphx::make_op("slice", {{"axes", {2}}, {"mode", "starts_ends_input"}}), | |
| input, | |
| starts, | |
| ends); | |
| mm->add_instruction(migraphx::make_op("slice", {{"axes", {2}}, {"mode", "starts_ends_input"}}), | |
| input, | |
| starts, | |
| ends); |
Comment on lines
537
to
+542
| auto slice_ins = m0.add_instruction( | ||
| migraphx::make_op("slice"), input, input_starts, input_ends, input_axes); | ||
| migraphx::make_op("slice", {{"mode", "starts_ends_axes_input"}}), | ||
| input, | ||
| input_starts, | ||
| input_ends, | ||
| input_axes); |
Contributor
There was a problem hiding this comment.
[format.py] reported by reviewdog 🐶
Suggested change
| auto slice_ins = m0.add_instruction( | |
| migraphx::make_op("slice"), input, input_starts, input_ends, input_axes); | |
| migraphx::make_op("slice", {{"mode", "starts_ends_axes_input"}}), | |
| input, | |
| input_starts, | |
| input_ends, | |
| input_axes); | |
| auto slice_ins = | |
| m0.add_instruction(migraphx::make_op("slice", {{"mode", "starts_ends_axes_input"}}), | |
| input, | |
| input_starts, | |
| input_ends, | |
| input_axes); |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
9 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Motivation
Technical Details
Changelog Category
Add a
CHANGELOG.mdentry for any option other thanNot ApplicableFollow the LLVM AI Tool Use Policy for contributions using AI.