Function run_modify_multi_axis_vec_empty_shape

src/opr/test/indexing.cpp:1219–1243

template <class Opr>
void run_modify_multi_axis_vec_empty_shape(
        const TensorShape& ishp, const TensorShape& vshp, const TensorShape& idx0,
        const TensorShape& idx1) {
    mgb_assert(ishp.ndim >= 4);
    mgb_assert(vshp.is_empty() && (idx0.is_empty() || idx1.is_empty()));
    using AI = opr::indexing::AxisIndexer;
    auto graph = ComputingGraph::make();
    HostTensorGenerator<> gen;
    HostTensorGenerator<dtype::Int32> gen_idx;
    auto host_x = gen(ishp), host_v = gen(vshp);
    auto x = opr::Host2DeviceCopy::make(*graph, host_x),
         v = opr::Host2DeviceCopy::make(*graph, host_v),
         idx_dynamic_shape = opr::MarkDynamicVar::make(
                 opr::ImmutableTensor::make(*graph, *gen_idx(idx0))),
         idx_static_shape = opr::ImmutableTensor::make(*graph, *gen_idx(idx1)),
         y = Opr::make(
                 x, v,
                 {AI::make_interval(-1, None, None, x.make_scalar(2)),
                  AI::make_index(1, idx_dynamic_shape),
                  AI::make_index(2, idx_static_shape)});
    HostTensorND host_y;
    auto func = graph->compile({make_callback_copy(y, host_y)});
    func->execute();
    MGB_ASSERT_TENSOR_EQ(*host_x, host_y);
}

}  // namespace