Function mean_abs_asymm

smallthinker/tests/test-backend-ops.cpp:198–223

maximum absolute asymmetry between a and b asymmetry: (a - b) / (a + b) This is more stable than relative error if one of the values fluctuates towards zero. n: number of values to compare. expected_vals: optional vector of expected values for a. If expected_vals is not empty, filter out all comparisons where a does not match any of the expected values. Needed for noncontinuous gradients where the

// expected_vals: optional vector of expected values for a. If expected_vals is not empty, filter out all comparisons where
//     a does not match any of the expected values. Needed for noncontinuous gradients where the numerical calculation can fail.
static double mean_abs_asymm(const float * a, const float * b, const size_t n, const std::vector<float> & expected_vals) {
    double sum = 0.0f;

    size_t nvalid = 0;
    for (size_t i = 0; i < n; i++) {
        if (!expected_vals.empty()) {
            bool matches_any = false;
            for (const float & ev : expected_vals) {
                if (fabsf(a[i] - ev) < 1e-3f) {
                    matches_any = true;
                    break;
                }
            }
            if (!matches_any) {
                continue;
            }
        }

        const float asymm = (a[i] - b[i]) / (a[i] + b[i]);

        sum += fabsf(asymm);
        nvalid++;
    }

    return sum/nvalid;
}

// utils for printing the variables of the test cases