NUMBA_DTYPES = ["int8", "int16", "int32", "int64", "uint8", "uint16", "uint32", "uint64", "float32", "float64", "float16"]

NUMBA_TO_NVSHMEM_TYPES = {
    "int8": "int8",
    "int16": "int16",
    "int32": "int32",
    "int64": "int64",
    "uint8": "uint8",
    "uint16": "uint16",
    "uint32": "uint32",
    "uint64": "uint64",
    "float32": "float",
    "float64": "double",
    "float16": "half",
    "bfloat16": "bfloat16",
}

NUMBA_TO_NVSHMEM_TYPES_AMO = {
    "int32": "int",
    "int64": "long",
    "uint32": "uint32",
    "uint64": "size",
    "float32": "float",
    "float64": "double",
}

