{
  "op": "ai.onnx.Mod",
  "tunableSpace": { "WORKGROUP_SIZE": [64, 128, 256] },
  "cases": [
    {
      "name": "8m_u32_scalar_div",
      "preset": "smoke",
      "inputs": {
        "a": { "dtype": "uint32", "shape": [8388608] },
        "b": { "dtype": "uint32", "shape": [1], "data": { "kind": "constant", "value": 17 } }
      },
      "outputs": { "c": { "dtype": "uint32", "shape": [8388608] } },
      "bench": { "primary": true, "metrics": [{ "type": "bandwidth", "value": "8388608 * 4 * 2" }] }
    },
    {
      "name": "1m_u32",
      "inputs": {
        "a": { "dtype": "uint32", "shape": [1048576] },
        "b": { "dtype": "uint32", "shape": [1], "data": { "kind": "constant", "value": 17 } }
      },
      "outputs": { "c": { "dtype": "uint32", "shape": [1048576] } }
    },
    {
      "name": "1m_f32_fmod_same_shape",
      "attrs": { "fmod": 1 },
      "inputs": {
        "a": { "dtype": "float32", "shape": [1048576] },
        "b": { "dtype": "float32", "shape": [1048576], "data": { "kind": "constant", "value": 3.7 } }
      },
      "outputs": { "c": { "dtype": "float32", "shape": [1048576] } }
    },
    {
      "name": "f32_broadcast_per_channel_fallback",
      "preset": "smoke",
      "vars": { "dtype": "float32" },
      "attrs": { "fmod": 1 },
      "inputs": {
        "a": { "dtype": "float32", "shape": [2048, 2049], "dist": "normal", "seed": 711, "scale": 2 },
        "b": { "dtype": "float32", "shape": [2049], "data": { "kind": "constant", "value": 3.0 } }
      },
      "outputs": { "c": { "dtype": "float32", "shape": [2048, 2049] } },
      "bench": { "metrics": [{ "type": "bandwidth", "value": "2048 * 2049 * 2 * dtypeBytes(args.dtype)" }] }
    },
    {
      "name": "1m_f16_fmod_same_shape",
      "preset": "smoke",
      "vars": { "dtype": "float16", "count": 1048576 },
      "attrs": { "fmod": 1 },
      "inputs": {
        "a": { "dtype": "float16", "shape": [1048576], "dist": "normal", "seed": 712, "scale": 8 },
        "b": { "dtype": "float16", "shape": [1048576], "data": { "kind": "constant", "value": 3.5 } }
      },
      "outputs": { "c": { "dtype": "float16", "shape": [1048576] } },
      "bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 3" }] }
    },
    {
      "name": "1m_i32_floor_mod_same_shape",
      "preset": "smoke",
      "vars": { "dtype": "int32", "count": 1048576 },
      "inputs": {
        "a": { "dtype": "int32", "shape": [1048576], "dist": "normal", "seed": 713, "scale": 1000 },
        "b": { "dtype": "int32", "shape": [1048576], "data": { "kind": "constant", "value": 7 } }
      },
      "outputs": { "c": { "dtype": "int32", "shape": [1048576] } },
      "bench": { "metrics": [{ "type": "bandwidth", "value": "args.count * dtypeBytes(args.dtype) * 3" }] }
    },
    {
      "name": "i32_floor_mod_broadcast_non_vec4_shape",
      "preset": "stress",
      "attrs": { "fmod": 0 },
      "inputs": {
        "a": { "dtype": "int32", "shape": [2049, 2051], "dist": "normal", "seed": 801, "scale": 1000 },
        "b": { "dtype": "int32", "shape": [2051], "data": { "kind": "constant", "value": 7 } }
      },
      "outputs": { "c": { "dtype": "int32", "shape": [2049, 2051] } },
      "bench": { "metrics": [{ "type": "bandwidth", "value": "2049 * 2051 * 4 * 2" }] }
    }
  ]
}