File size: 2,078 Bytes
aeefdfd
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
  "cases": [
    {
      "name": "1m_f32",
      "inputs": { "x": { "dtype": "float32", "shape": [1048576] } },
      "outputs": { "y": { "dtype": "float32", "shape": [1048576] } }
    },
    {
      "name": "tan-f32-1m-vec4",
      "preset": "smoke",
      "vars": { "dtype": "float32", "count": 1048576 },
      "inputs": {
        "x": { "shape": [1048576], "dtype": "float32", "dist": "uniform", "seed": 7101, "scale": 2.5, "offset": -1.25 }
      },
      "outputs": { "y": { "shape": [1048576], "dtype": "float32" } },
      "bench": {
        "primary": true,
        "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }]
      }
    },
    {
      "name": "tan-f16-1m-vec4",
      "preset": "smoke",
      "vars": { "dtype": "float16", "count": 1048576 },
      "inputs": {
        "x": { "shape": [1048576], "dtype": "float16", "dist": "uniform", "seed": 7102, "scale": 2.5, "offset": -1.25 }
      },
      "outputs": { "y": { "shape": [1048576], "dtype": "float16" } },
      "bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
    },
    {
      "name": "tan-f32-1m-plus-1-scalar-fallback",
      "preset": "smoke",
      "vars": { "dtype": "float32", "count": 1048577 },
      "inputs": {
        "x": { "shape": [1048577], "dtype": "float32", "dist": "uniform", "seed": 7103, "scale": 2.5, "offset": -1.25 }
      },
      "outputs": { "y": { "shape": [1048577], "dtype": "float32" } },
      "bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
    },
    {
      "name": "tan-f16-1m-plus-1-scalar-fallback",
      "preset": "smoke",
      "vars": { "dtype": "float16", "count": 1048577 },
      "inputs": {
        "x": { "shape": [1048577], "dtype": "float16", "dist": "uniform", "seed": 7104, "scale": 2.5, "offset": -1.25 }
      },
      "outputs": { "y": { "shape": [1048577], "dtype": "float16" } },
      "bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 2" }] }
    }
  ]
}