TheRock/rocm-libraries/dnn-providers/miopen-provider/MiopenActivationDescriptor.cpp

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

#include "MiopenActivationDescriptor.hpp"
#include "MiopenUtils.hpp"

#include <hipdnn_plugin_sdk/PluginException.hpp>

namespace miopen_plugin
{

MiopenActivationDescriptor::MiopenActivationDescriptor(
    const hipdnn_flatbuffers_sdk::data_objects::PointwiseAttributes& pointwiseAttrs)
{
    using namespace miopen_utils;

    ActivationParams params;
    HIPDNN_PREPEND_MESSAGE_ON_THROW(params = mapPointwiseModeToMiopenActivation(pointwiseAttrs),
                                    "MiopenActivationDescriptor: ");

    THROW_ON_MIOPEN_FAILURE(miopenCreateActivationDescriptor(&_descriptor));
    THROW_ON_MIOPEN_FAILURE(miopenSetActivationDescriptor(
        _descriptor, params.mode, params.alpha, params.beta, params.gamma));
}

MiopenActivationDescriptor::MiopenActivationDescriptor(MiopenActivationDescriptor&& other) noexcept
    : _descriptor(other._descriptor)
{
    other._descriptor = nullptr;
}

MiopenActivationDescriptor&
    MiopenActivationDescriptor::operator=(MiopenActivationDescriptor&& other) noexcept
{
    if(this == &other)
    {
        return *this;
    }

    if(_descriptor != nullptr)
    {
        LOG_ON_MIOPEN_FAILURE(miopenDestroyActivationDescriptor(_descriptor));
    }

    _descriptor = other._descriptor;
    other._descriptor = nullptr;
    return *this;
}

MiopenActivationDescriptor::~MiopenActivationDescriptor()
{
    if(_descriptor != nullptr)
    {
        LOG_ON_MIOPEN_FAILURE(miopenDestroyActivationDescriptor(_descriptor));
    }
}

miopenActivationDescriptor_t MiopenActivationDescriptor::activationDescriptor() const
{
    return _descriptor;
}

} // namespace miopen_plugin