Function generate_argument

src/generate.cpp:53–93

}

argument generate_argument(shape s, unsigned long seed, random_mode m)
{
    argument result;
    if(s.type() == shape::tuple_type)
    {
        const auto& sub_ss = s.sub_shapes();
        std::vector<argument> sub_args;
        std::transform(sub_ss.begin(), sub_ss.end(), std::back_inserter(sub_args), [&](auto ss) {
            return generate_argument(ss, seed, m);
        });

        result = argument(sub_args);
    }
    // special processing for non-computable type
    else if(not s.computable())
    {
        // NOTE: these values can be wrong (ex. not valid fp4x2)
        auto v = generate_tensor_data<uint8_t>(s, seed, m);
        result = {s, v};
    }
    else
    {
        s.visit_type([&](auto as) {
            // we use char type to store bool type internally, so bool_type
            // needs special processing to generate data
            if(s.type() == shape::bool_type)
            {
                auto v = generate_tensor_data<bool>(s, seed, m);
                result = {s, v};
            }
            else
            {
                using type = typename decltype(as)::type;
                auto v     = generate_tensor_data<type>(s, seed, m);
                result     = {s, v};
            }
        });
    }

    return result;
}

literal generate_literal(shape s, unsigned long seed)
{