},
}
-backends = ['CPU', 'CUDA', 'QuantizedCPU']
-densities = ['Dense', 'Sparse', 'Mkldnn'] # TODO: layout instead of densities?
+
def backend_to_devicetype(backend):
if backend == 'QuantizedCPU':
return 'CPU'
return backend
+backends = ['CPU', 'CUDA', 'QuantizedCPU']
+densities = ['Dense', 'Sparse', 'Mkldnn'] # TODO: layout instead of densities?
+
quantized_backends = ['QuantizedCPU']
extension_backends = ['MSNPU', 'XLA']
] + quantized_scalar_types
-
# shared environment for non-derived base classes Type.h Tensor.h Storage.h
top_env = {
'cpu_type_registrations': [],
continue
if backend in quantized_backends:
if density == 'Dense' and scalar_type in quantized_scalar_types:
- yield (backend, density, scalar_type)
+ yield (backend, density, scalar_type)
else:
continue
else: