This is an archive of the discontinued LLVM Phabricator instance.

Show First 20 Lines • Show All 347 Lines • ▼ Show 20 Lines	class Format:

def rank(self) -> int:		def rank(self) -> int:
"""Returns the number of dimensions represented by the format."""		"""Returns the number of dimensions represented by the format."""
return self.format_pack.rank()		return self.format_pack.rank()

def get_permutation_and_sparsity(self) -> Tuple[np.ndarray, np.ndarray]:		def get_permutation_and_sparsity(self) -> Tuple[np.ndarray, np.ndarray]:
"""Constructs the numpy arrays for the permutation and sparsity."""		"""Constructs the numpy arrays for the permutation and sparsity."""
perm = np.array(self.ordering.ordering, dtype=np.ulonglong)		perm = np.array(self.ordering.ordering, dtype=np.ulonglong)
a = [0 if s == ModeFormat.DENSE else 1 for s in self.format_pack.formats]		a = [f.value for f in self.format_pack.formats]
sparse = np.array(a, dtype=np.uint8)		sparse = np.array(a, dtype=np.uint8)
return (perm, sparse)		return (perm, sparse)

def mlir_tensor_attr(self) -> Optional[sparse_tensor.EncodingAttr]:		def mlir_tensor_attr(self) -> Optional[sparse_tensor.EncodingAttr]:
"""Constructs the MLIR attributes for the tensor format."""		"""Constructs the MLIR attributes for the tensor format."""
order = (		order = (
range(self.rank()) if		range(self.rank()) if
(self.ordering is None) else self.ordering.ordering)		(self.ordering is None) else self.ordering.ordering)
▲ Show 20 Lines • Show All 1,831 Lines • Show Last 20 Lines

[mlir][sparse][taco] Avoid hard-coding DimLevelType values.ClosedPublic