File: EqualityComparers.cs

package info (click to toggle)
onnxruntime 1.21.0%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 333,732 kB
  • sloc: cpp: 3,153,079; python: 179,219; ansic: 109,131; asm: 37,791; cs: 34,424; perl: 13,070; java: 11,047; javascript: 6,330; pascal: 4,126; sh: 3,277; xml: 598; objc: 281; makefile: 59
file content (80 lines) | stat: -rw-r--r-- 2,001 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
using Microsoft.ML.OnnxRuntime.Tensors;
using System;
using System.Collections.Generic;

namespace Microsoft.ML.OnnxRuntime.Tests
{
    internal class FloatComparer : IEqualityComparer<float>
    {
        private float atol = 1e-3f;
        private float rtol = 1.7e-2f;

        public bool Equals(float x, float y)
        {
            return Math.Abs(x - y) <= (atol + rtol * Math.Abs(y));
        }
        public int GetHashCode(float x)
        {
            return x.GetHashCode();
        }
    }

    internal class DoubleComparer : IEqualityComparer<double>
    {
        private double atol = 1e-3;
        private double rtol = 1.7e-2;

        public bool Equals(double x, double y)
        {
            return Math.Abs(x - y) <= (atol + rtol * Math.Abs(y));
        }
        public int GetHashCode(double x)
        {
            return x.GetHashCode();
        }
    }

    internal class ExactComparer<T> : IEqualityComparer<T>
    {
        public bool Equals(T x, T y)
        {
            return x.Equals(y);
        }
        public int GetHashCode(T x)
        {
            return x.GetHashCode();
        }
    }

    /// <summary>
    /// Use it to compare Float16
    /// </summary>
    internal class Float16Comparer : IEqualityComparer<Float16>
    {
        public ushort tolerance = 0;
        public bool Equals(Float16 x, Float16 y)
        {
            return Math.Abs(x.value - y.value) <= (tolerance + y.value);
        }
        public int GetHashCode(Float16 x)
        {
            return x.GetHashCode();
        }
    }

    /// <summary>
    /// Use it to compare Bloat16
    /// </summary>
    internal class BFloat16Comparer : IEqualityComparer<BFloat16>
    {
        public ushort tolerance = 0;
        public bool Equals(BFloat16 x, BFloat16 y)
        {
            return Math.Abs(x.value - y.value) <= (tolerance + y.value);
        }
        public int GetHashCode(BFloat16 x)
        {
            return x.GetHashCode();
        }
    }
}