Use immutabledict for global constants in python scripts

PiperOrigin-RevId: 914562268

Author: fbarchard

tools/generate-vbinary-test.py

@@ -11,6 +11,7 @@
 import re
 import sys
 import yaml
+from immutabledict import immutabledict
 
 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
 import xngen
@@ -44,7 +45,7 @@
 )
 parser.set_defaults(defines=list())
 
-OP_TYPES = {
+OP_TYPES = immutabledict({
     "vadd": "Add",
     "vaddc": "Add",
     "vcopysign": "CopySign",
@@ -71,7 +72,7 @@
     "vprelu": "Prelu",
     "vpreluc": "Prelu",
     "vrpreluc": "RPrelu",
-}
+})
 
 BINOP_TEST_TEMPLATE = """
 #define XNN_UKERNEL(arch_flags, ukernel, batch_tile, vector_tile, datatype, params_type, init_params)

tools/generate-vunary-test.py

@@ -8,9 +8,10 @@
 import math
 import os
 import sys
-import types
 from typing import NamedTuple
 
+from immutabledict import immutabledict
+
 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
 import xngen
 import xnncommon
@@ -41,7 +42,7 @@ class SpecialValues(NamedTuple):
     expected_outputs: str
     tolerance_ulp: int
 
-OP_TYPES = {
+OP_TYPES = immutabledict({
     "vabs": "Abs",
     "vapproxgelu": "ApproxGELU",
     "vclamp": "Clamp",
@@ -63,11 +64,11 @@ class SpecialValues(NamedTuple):
     "vsqr": "Square",
     "vsqrt": "SquareRoot",
     "vtanh": "TanH",
-}
+})
 
 PARAMS_TYPES = ["Clamp", "ELU", "LeakyReLU"]
 
-SPECIAL_VALUES_BY_OP_TYPE_F32 = types.MappingProxyType({
+SPECIAL_VALUES_BY_OP_TYPE_F32 = immutabledict({
     "SquareRoot": SpecialValues(
         num_elements=4,
         inputs="{0.0f, -0.0f, 1.0f, -1.0f}",
@@ -127,7 +128,7 @@ class SpecialValues(NamedTuple):
 })
 
 
-SPECIAL_VALUES_BY_OP_TYPE_F16 = types.MappingProxyType({
+SPECIAL_VALUES_BY_OP_TYPE_F16 = immutabledict({
     "Log": SpecialValues(
         num_elements=4,
         inputs="{1.0f, -1.0f, 0.0f, -0.0f}",

tools/update-microkernels.py

@@ -10,6 +10,8 @@
 import os
 import re
 import sys
+from immutabledict import immutabledict
+
 
 parser = argparse.ArgumentParser(
     description='Utility for re-generating microkernel lists'
@@ -81,12 +83,12 @@
     'wasmsimd',
 })
 
-_ISA_MAP = {
+_ISA_MAP = immutabledict({
     'wasmblendvps': 'wasmrelaxedsimd',
     'wasmpshufb': 'wasmrelaxedsimd',
     'wasmsdot': 'wasmrelaxedsimd',
     'wasmusdot': 'wasmrelaxedsimd',
-}
+})
 
 _ARCH_LIST = frozenset({
     'aarch32',
@@ -101,12 +103,12 @@
     r'\bxnn_(?:[a-z0-9]+(?:_[a-z0-9]+)*)_ukernel(?:_[a-z0-9]+)*__(?:[a-z0-9]+(?:_[a-z0-9]+)*)\b'
 )
 
-_VERIFICATION_IGNORE_SUBDIRS = {
+_VERIFICATION_IGNORE_SUBDIRS = frozenset({
     os.path.join('src', 'qs8-requantization'),
     os.path.join('src', 'qu8-requantization'),
     os.path.join('src', 'reference'),
     os.path.join('src', 'xnnpack', 'simd'),
-}
+})
 
 
 def overwrite_if_changed(filepath, content):

tools/xnncommon.py

@@ -6,6 +6,8 @@
 
 import codecs
 import os
+from immutabledict import immutabledict
+
 
 
 def _indent(text):
@@ -23,7 +25,7 @@ def _remove_duplicate_newlines(text):
   return "\n".join(filtered_lines)
 
 
-_ARCH_TO_MACRO_MAP = {
+_ARCH_TO_MACRO_MAP = immutabledict({
     "aarch32": "XNN_ARCH_ARM",
     "aarch64": "XNN_ARCH_ARM64",
     "x86-32": "XNN_ARCH_X86",
@@ -38,11 +40,11 @@ def _remove_duplicate_newlines(text):
     "wasmsimd32": "XNN_ARCH_WASMSIMD",
     "wasmrelaxedsimd32": "XNN_ARCH_WASMRELAXEDSIMD",
     "wasmrelaxedsimdfp16": "XNN_ARCH_WASMRELAXEDSIMDFP16",
-}
+})
 
 # Mapping from ISA extension to macro guarding build-time enabled/disabled
 # status for the ISA. Only ISAs that can be enabled/disabled have an entry.
-_ISA_TO_MACRO_MAP = {
+_ISA_TO_MACRO_MAP = immutabledict({
     "fp16arith": "XNN_ENABLE_ARM_FP16_SCALAR",
     "neonfp16arith": "XNN_ENABLE_ARM_FP16_VECTOR",
     "neonbf16": "XNN_ENABLE_ARM_BF16",
@@ -75,9 +77,9 @@ def _remove_duplicate_newlines(text):
     "avx512fp16": "XNN_ENABLE_AVX512FP16",
     "avx512bf16": "XNN_ENABLE_AVX512BF16",
     "hvx": "XNN_ENABLE_HVX",
-}
+})
 
-_ISA_TO_ARCH_MAP = {
+_ISA_TO_ARCH_MAP = immutabledict({
     "armsimd32": ["aarch32"],
     "fp16arith": ["aarch32", "aarch64"],
     "neon": ["aarch32", "aarch64"],
@@ -124,9 +126,9 @@ def _remove_duplicate_newlines(text):
     "wasmsdot": ["wasmrelaxedsimd"],
     "wasmusdot": ["wasmrelaxedsimd"],
     "wasmblendvps": ["wasmrelaxedsimd"],
-}
+})
 
-_ISA_TO_ARCH_FLAGS_MAP = {
+_ISA_TO_ARCH_FLAGS_MAP = immutabledict({
     "armsimd32": "xnn_arch_arm_v6",
     "fp16arith": "xnn_arch_arm_fp16_arith",
     "neon": "xnn_arch_arm_neon",
@@ -168,7 +170,7 @@ def _remove_duplicate_newlines(text):
     "wasmsdot": "xnn_arch_wasm_sdot",
     "wasmusdot": "xnn_arch_wasm_usdot",
     "wasmblendvps": "xnn_arch_wasm_blendvps",
-}
+})
 
 
 def isa_hierarchy_map():
@@ -268,7 +270,7 @@ def postprocess_test_case(test_case, arch, isa, assembly=False):
     "hvx",
 ]
 
-_ISA_HIERARCHY_MAP = {isa: v for v, isa in enumerate(_ISA_HIERARCHY)}
+_ISA_HIERARCHY_MAP = immutabledict({isa: v for v, isa in enumerate(_ISA_HIERARCHY)})
 
 
 def overwrite_if_changed(filepath, content):