Function check_rng_basic

imperative/src/test/rng.cpp:9–33

template <typename Op, typename... Args>
void check_rng_basic(Args&&... args) {
    for (auto&& tshape : {TensorShape{2, 3, 4, 5}, {3, 4, 5, 6}, {2333}})
        for (auto&& cn : {CompNode::load("xpu0"), CompNode::load("xpu1")}) {
            Handle h = new_handle(cn, 123);
            auto op = Op::make(std::forward<Args>(args)..., h);
            DeviceTensorND tshape_dev;
            cg::copy_shape_to_tensor_value(tshape_dev, tshape);
            SmallVector<TensorPtr> inputs = {Tensor::make(tshape_dev)};
            SmallVector<LogicalTensorDesc> input_descs;
            input_descs.push_back(
                    {inputs[0]->layout(), inputs[0]->comp_node(),
                     inputs[0]->dev_tensor()});
            auto [output_descs, validated] =
                    OpDef::infer_output_attrs_fallible(*op, input_descs);
            auto outputs = OpDef::apply_on_physical_tensor(
                    *op, inputs, output_descs, validated);
            ASSERT_TRUE(outputs[0]->layout().eq_shape(tshape));
            ASSERT_TRUE(cn == outputs[0]->comp_node());
            // sync before delete handle
            for (auto&& p : outputs) {
                p->get_value();
            }
            delete_handle(h);
        }
}

template <typename Op, typename... Args>
void check_rng_with_input_basic(