gemv.json 385 Bytes
Newer Older
sunzhq2's avatar
init  
sunzhq2 committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
{
  "operator": "gemv",
  "iterations": 100,
  "input_shape_groups": [
    {
      "M": [1],
      "K": [16, 32, 64, 128, 256, 512, 1024, 2048, 4096, 8192],
      "N": [4096, 8192]
    },
    {
      "M": [1],
      "K": [4096, 8192],
      "N": [16, 32, 64, 128, 256, 512, 1024, 2048, 4096, 8192]
    }
  ],
  "dtype": [
    "float32",
    "bfloat16",
    "float16", 
    "int8"
  ]
}