scipy/get_cpu_flags.py at wheel-variants · wheelnext/scipy · 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
#!/usr/bin/env python3

import argparse


def main() -> None:
    argp = argparse.ArgumentParser()
    argp.add_argument("--arch", required=True, choices=["aarch64", "x86_64"])
    argp.add_argument("--language", required=True)
    argp.add_argument("--compiler-name", required=True)
    argp.add_argument("--compiler-version", required=True)
    argp.add_argument("--variant", required=True)

    args = argp.parse_args()

    if args.arch == "aarch64":
        from provider_variant_aarch64.plugin import AArch64Plugin
        plugin = AArch64Plugin()
    elif args.arch == "x86_64":
        from provider_variant_x86_64.plugin import X8664Plugin
        plugin = X8664Plugin()

    vprops = [
        argparse.Namespace(
            namespace=args.arch,
            feature="level" if args.arch == "x86_64" else "version",
            value=args.variant,
        )
    ]
    print(
        " ".join(
            plugin.get_compiler_flags(
                args.language, args.compiler_name, args.compiler_version, vprops
            )
        )
    )


if __name__ == "__main__":
    main()