kernel_zoo/schemas/desc.json at main · MetaInfer/kernel_zoo · GitHub

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
{
  "$schema": "https://json-schema.org/draft/2020-12/schema",
  "$id": "https://kernel_zoo/schemas/desc.json",
  "title": "kernel_package desc.yaml",
  "type": "object",
  "required": [
    "schema_version",
    "name",
    "category",
    "summary",
    "description",
    "io_signature",
    "quantization",
    "reference_sources",
    "status"
  ],
  "properties": {
    "schema_version": {"type": "string", "minLength": 1},
    "name": {"type": "string", "minLength": 1},
    "category": {
      "type": "object",
      "required": ["arch", "op_class", "sub_class", "quant"],
      "additionalProperties": false,
      "properties": {
        "arch": {"type": "string", "minLength": 1},
        "op_class": {"type": "string", "minLength": 1},
        "sub_class": {"type": "string", "minLength": 1},
        "quant": {"type": "string", "minLength": 1}
      }
    },
    "summary": {"type": "string", "minLength": 1, "maxLength": 300},
    "description": {"type": "string", "minLength": 1},
    "io_signature": {
      "type": "array",
      "minItems": 1,
      "items": {
        "type": "object",
        "required": ["name", "kind", "dtype", "shape"],
        "additionalProperties": false,
        "properties": {
          "name": {"type": "string", "minLength": 1},
          "kind": {"enum": ["in", "out", "inout"]},
          "dtype": {"type": "string", "minLength": 1},
          "shape": {
            "type": "array",
            "items": {"type": ["integer", "string"], "minLength": 1}
          },
          "shape_desc": {"type": "string"},
          "semantics": {"type": "string"}
        }
      }
    },
    "quantization": {
      "type": "object",
      "required": ["scheme"],
      "additionalProperties": false,
      "properties": {
        "scheme": {"type": "string", "minLength": 1},
        "granularity": {"enum": ["per_tensor", "per_block", "per_channel"]},
        "block_size": {"type": "integer", "minimum": 1},
        "group_size": {"type": "integer", "minimum": 1}
      }
    },
    "reference_sources": {
      "type": "array",
      "minItems": 1,
      "items": {"type": "string", "minLength": 1}
    },
    "status": {"enum": ["normal", "frozen", "deprecated"]},
    "min_arch_feature": {"type": "string"},
    "tags": {"type": "array", "items": {"type": "string"}},
    "owner": {"type": "string"}
  }
}