File size: 1,899 Bytes
ea833c8 | 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 | {
"op": "ai.onnx.BitwiseAnd",
"cases": [
{
"name": "u32_16m",
"preset": "stress",
"inputs": {
"a": { "dtype": "uint32", "shape": [16777216], "data": { "kind": "constant", "value": 305419896 } },
"b": { "dtype": "uint32", "shape": [16777216], "data": { "kind": "constant", "value": 252645135 } }
},
"outputs": { "c": { "dtype": "uint32", "shape": [16777216] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "16777216 * 4 * 3" }] }
},
{
"name": "u32_1m",
"preset": "smoke",
"inputs": {
"a": { "dtype": "uint32", "shape": [1048576], "data": { "kind": "constant", "value": 305419896 } },
"b": { "dtype": "uint32", "shape": [1048576], "data": { "kind": "constant", "value": 252645135 } }
},
"outputs": { "c": { "dtype": "uint32", "shape": [1048576] } },
"bench": { "primary": true, "metrics": [{ "type": "bandwidth", "value": "1048576 * 4 * 3" }] }
},
{
"name": "u32_4096x4096_bcast_2d",
"inputs": {
"a": { "dtype": "uint32", "shape": [4096, 1], "data": { "kind": "constant", "value": 305419896 } },
"b": { "dtype": "uint32", "shape": [1, 4096], "data": { "kind": "constant", "value": 252645135 } }
},
"outputs": { "c": { "dtype": "uint32", "shape": [4096, 4096] } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "4096 * 4096 * 4" }] }
},
{
"name": "u32_same_shape_odd_numel",
"preset": "stress",
"inputs": {
"a": { "dtype": "uint32", "shape": [16777217], "dist": "uniformBits", "seed": 1 },
"b": { "dtype": "uint32", "shape": [16777217], "dist": "uniformBits", "seed": 2 }
},
"outputs": { "c": { "dtype": "uint32", "shape": [16777217], "dist": "empty" } },
"bench": { "metrics": [{ "type": "bandwidth", "value": "16777217 * 4 * 3" }] }
}
]
}
|