GeezerVison is showing github.com — arrayfire-py/arrayfire/array_api/_dtypes.py at master · arrayfire/arrayfire-py

Expand file tree

_dtypes.py

Copy path

More file actions

Latest commit

History

172 lines (161 loc) · 4.94 KB

_dtypes.py

Copy path

File metadata and controls

172 lines (161 loc) · 4.94 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

from __future__ import annotations

import ctypes

__all__ = [

"all_dtypes",

"boolean_dtypes",

"real_floating_dtypes",

"floating_dtypes",

"complex_floating_dtypes",

"integer_dtypes",

"signed_integer_dtypes",

"unsigned_integer_dtypes",

"integer_or_boolean_dtypes",

"real_numeric_dtypes",

"numeric_dtypes",

"dtype_categories",

# OG

"bool",

"complex64",

"complex128",

"float32",

"float64",

"int8",

"int16",

"int32",

"int64",

"uint8",

"uint16",

"uint32",

"uint64",

]

from arrayfire import bool

from arrayfire import complex32 as afcomplex32

from arrayfire import complex64 as afcomplex64

from arrayfire import float16, float32, float64, int16, int32, int64, uint8, uint16, uint32, uint64

from arrayfire.dtypes import Dtype

int8 = Dtype("int8", "b8", ctypes.c_char, "int8", 4) # HACK int8 is not supported in AF. Selecting same as b8

complex64 = afcomplex32

complex128 = afcomplex64

all_dtypes = (

int8,

int16,

int32,

int64,

uint8,

uint16,

uint32,

uint64,

float16,

float32,

float64,

complex64,

complex128,

bool,

)

boolean_dtypes = (bool,)

real_floating_dtypes = (float16, float32, float64)

floating_dtypes = (float16, float32, float64, complex64, complex128)

complex_floating_dtypes = (complex64, complex128)

integer_dtypes = (int8, int16, int32, int64, uint8, uint16, uint32, uint64)

signed_integer_dtypes = (int8, int16, int32, int64)

unsigned_integer_dtypes = (uint8, uint16, uint32, uint64)

integer_or_boolean_dtypes = boolean_dtypes + integer_dtypes

real_numeric_dtypes = real_floating_dtypes + integer_dtypes

numeric_dtypes = floating_dtypes + integer_dtypes + complex_floating_dtypes

dtype_categories = {

"all": all_dtypes,

"real numeric": real_numeric_dtypes,

"numeric": numeric_dtypes,

"integer": integer_dtypes,

"integer or boolean": integer_or_boolean_dtypes,

"boolean": boolean_dtypes,

"real floating-point": floating_dtypes,

"complex floating-point": complex_floating_dtypes,

"floating-point": floating_dtypes,

}

# Note: the spec defines a restricted type promotion table compared to NumPy.

# In particular, cross-kind promotions like integer + float or boolean +

# integer are not allowed, even for functions that accept both kinds.

# Additionally, NumPy promotes signed integer + uint64 to float64, but this

# promotion is not allowed here. To be clear, Python scalar int objects are

# allowed to promote to floating-point dtypes, but only in array operators

# (see Array._promote_scalar) method in _array_object.py.

_promotion_table = {

(int8, int8): int8,

(int8, int16): int16,

(int8, int32): int32,

(int8, int64): int64,

(int16, int8): int16,

(int16, int16): int16,

(int16, int32): int32,

(int16, int64): int64,

(int32, int8): int32,

(int32, int16): int32,

(int32, int32): int32,

(int32, int64): int64,

(int64, int8): int64,

(int64, int16): int64,

(int64, int32): int64,

(int64, int64): int64,

(uint8, uint8): uint8,

(uint8, uint16): uint16,

(uint8, uint32): uint32,

(uint8, uint64): uint64,

(uint16, uint8): uint16,

(uint16, uint16): uint16,

(uint16, uint32): uint32,

(uint16, uint64): uint64,

(uint32, uint8): uint32,

(uint32, uint16): uint32,

(uint32, uint32): uint32,

(uint32, uint64): uint64,

(uint64, uint8): uint64,

(uint64, uint16): uint64,

(uint64, uint32): uint64,

(uint64, uint64): uint64,

(int8, uint8): int16,

(int8, uint16): int32,

(int8, uint32): int64,

(int16, uint8): int16,

(int16, uint16): int32,

(int16, uint32): int64,

(int32, uint8): int32,

(int32, uint16): int32,

(int32, uint32): int64,

(int64, uint8): int64,

(int64, uint16): int64,

(int64, uint32): int64,

(uint8, int8): int16,

(uint16, int8): int32,

(uint32, int8): int64,

(uint8, int16): int16,

(uint16, int16): int32,

(uint32, int16): int64,

(uint8, int32): int32,

(uint16, int32): int32,

(uint32, int32): int64,

(uint8, int64): int64,

(uint16, int64): int64,

(uint32, int64): int64,

(float32, float32): float32,

(float32, float64): float64,

(float64, float32): float64,

(float64, float64): float64,

(complex64, complex64): complex64,

(complex64, complex128): complex128,

(complex128, complex64): complex128,

(complex128, complex128): complex128,

(float32, complex64): complex64,

(float32, complex128): complex128,

(float64, complex64): complex128,

(float64, complex128): complex128,

(complex64, float32): complex64,

(complex64, float64): complex128,

(complex128, float32): complex128,

(complex128, float64): complex128,

(bool, bool): bool,

}

def promote_types(type1: Dtype, type2: Dtype) -> Dtype:

if (type1, type2) in _promotion_table:

return _promotion_table[type1, type2]

raise TypeError(f"{type1} and {type2} cannot be type promoted together")