TheRock/rocm-libraries/dnn-providers/hipblaslt-provider/tests/mocks/MockHipdnnEnginePluginExecutionContext.hpp

// Copyright © Advanced Micro Devices, Inc., or its affiliates.
// SPDX-License-Identifier:  MIT

#pragma once

#include <memory>

#include "mocks/MockPlan.hpp"

#include "HipdnnEnginePluginExecutionContext.hpp"

struct MockHipdnnEnginePluginExecutionContext : public HipdnnEnginePluginExecutionContext
{
    MockHipdnnEnginePluginExecutionContext()
        : mockPlan(std::make_unique<hipblaslt_plugin::MockPlan>())
    {
    }

    hipblaslt_plugin::IPlan& plan() const override
    {
        return *mockPlan;
    }

    std::unique_ptr<hipblaslt_plugin::MockPlan> mockPlan;
};