Method FlashSrGraph

src/framework/audio/flashsr.cpp:323–361

class FlashSrGraph {
public:
    FlashSrGraph(const FlashSrWeights & weights, int64_t input_samples)
        : weights_(weights),
          input_samples_(input_samples) {
        if (input_samples_ <= 0) {
            throw std::runtime_error("FlashSR graph input length must be positive");
        }
        ggml_init_params params{64ull * 1024ull * 1024ull, nullptr, true};
        ctx_.reset(ggml_init(params));
        if (ctx_ == nullptr) {
            throw std::runtime_error("failed to initialize FlashSR GGML context");
        }
        core::ModuleBuildContext build_ctx{ctx_.get(), "flashsr", weights.backend_type};
        auto input = core::make_tensor(build_ctx, GGML_TYPE_F32, core::TensorShape::from_dims({1, 1, input_samples_}));
        input_ = input.tensor;
        auto x = conv1d(build_ctx, input, weights.conv_pre, 3, 1);
        x = modules::Interpolate1dModule({input_samples_ * 3, modules::Interpolate1dMode::Linear}).build(build_ctx, x);
        auto xs = resblock(build_ctx, x, weights.resblock2, weights, 11);
        auto xs0 = resblock(build_ctx, x, weights.resblock0, weights, 3);
        xs = core::wrap_tensor(ggml_scale(build_ctx.ggml, modules::AddModule().build(build_ctx, xs, xs0).tensor, 0.5f), xs.shape, GGML_TYPE_F32);
        xs = activation1d(build_ctx, xs, weights.activation_post, weights);
        auto output = conv1d(build_ctx, xs, weights.conv_post, 3, 1);
        output = core::wrap_tensor(ggml_tanh(build_ctx.ggml, output.tensor), output.shape, GGML_TYPE_F32);
        output_ = output.tensor;
        ggml_set_output(output_);
        graph_ = ggml_new_graph_custom(ctx_.get(), 65536, false);
        ggml_build_forward_expand(graph_, output_);
        gallocr_ = ggml_gallocr_new(ggml_backend_get_default_buffer_type(weights.backend.get()));
        if (gallocr_ == nullptr ||
            !ggml_gallocr_reserve(gallocr_, graph_) ||
            !ggml_gallocr_alloc_graph(gallocr_, graph_)) {
            throw std::runtime_error("failed to allocate FlashSR GGML graph");
        }
        if (core::uses_host_graph_plan(weights.backend.get())) {
            plan_ = core::create_backend_graph_plan_if_host(weights.backend.get(), graph_);
            if (plan_ == nullptr) {
                throw std::runtime_error("failed to create FlashSR graph plan");
            }
        }
    }

    ~FlashSrGraph() {
        core::release_backend_graph_resources(weights_.backend.get(), graph_);