Maximumgrad
=================

计算逐元素 Maximum 操作的梯度。该算子是 Maximum 算子的反向传播部分。梯度 dy 将被路由到在前向传播中值较大的那个输入。

.. math::

    \text{dx0}_i = \begin{cases}
    \text{dy}_i, & \text{if } \text{Input0}_i > \text{Input1}_i \\
    0, & \text{otherwise}
    \end{cases}

    \text{dx1}_i = \begin{cases}
        \text{dy}_i, & \text{if } \text{Input1}_i \ge \text{Input0}_i \\
        0, & \text{otherwise}
    \end{cases}

输入：
    - **Input0** - 前向传播时的第一个输入数据地址。
    - **Input1** - 前向传播时的第二个输入数据地址。
    - **dy** - 后续层反向传播回来的梯度数据地址。
    - **params** - 参数数组：
        - **Input0_dims** - Input0 的维度信息数组。
        - **Input1_dims** - Input1 的维度信息数组。
        - **dy_dims** - dy 的维度信息数组。
        - **num_dims** - 输入张量的维度数量。
    - **core_mask** - 核掩码（仅共享存储版本需要）。

输出：
    - **dx0** - 计算出的关于 Input0 的梯度地址。
    - **dx1** - 计算出的关于 Input1 的梯度地址。

支持平台：
    ``FT78NE``
    ``MT7004``

.. note::
    - FT78NE 支持fp32
    - MT7004 支持fp16, fp32

**共享存储版本:**

.. c:function:: void hp_maximum_grad_s(half* Input0, half* Input1, half* dy, half* dx0, half* dx1, long long *params, int core_mask)
.. c:function:: void fp_maximum_grad_s(float* Input0, float* Input1, float* dy, float* dx0, float* dx1, long long *params, int core_mask)

**C调用示例：**

.. code-block:: c
    :linenos:
    :emphasize-lines: 17

    //FT78NE示例
    #include <stdio.h>
    #include <maximumgrad.h> // 假设头文件名为 maximumgrad.h

    int main(int argc, char* argv[]) {
        // 假设在DDR空间，且形状相同
        float *input0 = (float *)0xA0000000;
        float *input1 = (float *)0xA1000000;
        float *dy = (float *)0xA2000000;
        float *dx0 = (float *)0xB0000000;
        float *dx1 = (float *)0xB1000000;

        int dims[] = {4, 256};
        int num_dims = 2;
        int core_mask = 0xff;

        long long params[6];
        params[0] = (long long)dims; // Input0_dims
        params[1] = (long long)dims; // Input1_dims
        params[2] = (long long)dims; // dy_dims
        params[3] = (long long)num_dims; // num_dims    

        fp_maximum_grad_s(input0, input1, dy, dx0, dx1, core_mask);
        return 0;
    }

**私有存储版本:**

.. c:function:: void hp_maximum_grad_p(half* Input0, half* Input1, half* dy, long long *params, half* dx0, half* dx1)
.. c:function:: void fp_maximum_grad_p(float* Input0, float* Input1, float* dy, long long *params, float* dx0, float* dx1)

**C调用示例：**

.. code-block:: c
    :linenos:
    :emphasize-lines: 16

    //FT78NE示例
    #include <stdio.h>
    #include <maximumgrad.h> // 假设头文件名为 maximumgrad.h

    int main(int argc, char* argv[]) {
        // 假设在L2空间，且形状相同
        float *input0 = (float *)0x10000000;
        float *input1 = (float *)0x11000000;
        float *dy = (float *)0x12000000;
        float *dx0 = (float *)0x13000000;
        float *dx1 = (float *)0x14000000;

        int dims[] = {4, 256};
        int num_dims = 2;

        long long params[6];
        params[0] = (long long)dims; // Input0_dims
        params[1] = (long long)dims; // Input1_dims
        params[2] = (long long)dims; // dy_dims
        params[3] = (long long)num_dims; // num_dims 

        fp_maximum_grad_p(input0, input1, dy, params, dx0, dx1);
        return 0;
    }