cranelift-codegen 0.19.0

Low-level code generator library
Documentation 
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

"""
Generate sources with type info.

This generates a `types.rs` file which is included in
`lib/codegen/ir/types.rs`. The file provides constant definitions for the most
commonly used types, including all of the scalar types.

This ensures that Python and Rust use the same type numbering.
"""
from __future__ import absolute_import
import srcgen
from cdsl.types import ValueType
import base.types  # noqa

try:
    from typing import Iterable  # noqa
except ImportError:
    pass


def emit_type(ty, fmt):
    # type: (ValueType, srcgen.Formatter) -> None
    """
    Emit a constant definition of a single value type.
    """
    name = ty.name.upper()
    fmt.doc_comment(ty.__doc__)
    fmt.line(
            'pub const {}: Type = Type({:#x});'
            .format(name, ty.number))
    fmt.line()


def emit_vectors(bits, fmt):
    # type: (int, srcgen.Formatter) -> None
    """
    Emit definition for all vector types with `bits` total size.
    """
    size = bits // 8
    for ty in ValueType.all_lane_types:
        mb = ty.membytes
        if mb == 0 or mb >= size:
            continue
        emit_type(ty.by(size // mb), fmt)


def emit_types(fmt):
    # type: (srcgen.Formatter) -> None
    for spec in ValueType.all_special_types:
        emit_type(spec, fmt)
    for ty in ValueType.all_lane_types:
        emit_type(ty, fmt)
    # Emit vector definitions for common SIMD sizes.
    emit_vectors(64, fmt)
    emit_vectors(128, fmt)
    emit_vectors(256, fmt)
    emit_vectors(512, fmt)


def generate(out_dir):
    # type: (str) -> None
    fmt = srcgen.Formatter()
    emit_types(fmt)
    fmt.update_file('types.rs', out_dir)