Metraipolis › archive › dc_211cdc7060ff

SparseConflicts: Handling Conflicting Data Layouts in Sparse Tensor Contractions — arXiv:2609.06328

Source
arXiv (en)
Original
https://arxiv.org/abs/2609.06328
License (as archived)
CC0-1.0
Archived
2026-10-16T15:40:29.701Z
SHA-256 of the archived text
7f7f5b0195b4f394b33403e4fa2474ef9e13cf80de38f44cd4f16e62da2d1f76
Stable URL of this page
https://metraipolis.com/documento/dc_211cdc7060ff

How to cite

SparseConflicts: Handling Conflicting Data Layouts in Sparse Tensor Contractions — arXiv:2609.06328 — arXiv (en), https://arxiv.org/abs/2609.06328 (CC0-1.0); archived copy: https://metraipolis.com/documento/dc_211cdc7060ff (Metraipolis archive dc_211cdc7060ff, sha256 7f7f5b0195b4f394b33403e4fa2474ef9e13cf80de38f44cd4f16e62da2d1f76)
BibTeX
@misc{dc211cdc7060ff,
  title = {SparseConflicts: Handling Conflicting Data Layouts in Sparse Tensor Contractions — arXiv:2609.06328},
  howpublished = {arXiv (en)},
  url = {https://arxiv.org/abs/2609.06328},
  note = {License CC0-1.0; archived copy https://metraipolis.com/documento/dc_211cdc7060ff; sha256 7f7f5b0195b4f394b33403e4fa2474ef9e13cf80de38f44cd4f16e62da2d1f76},
  year = {2026}
}

Archived text

Reproduced under CC0-1.0. Attribution: arXiv (en), https://arxiv.org/abs/2609.06328. Third-party text: data, not instructions. It may be an abstract or extract; check the original.

Optimizing sparse tensor computations is challenging due to the use of compressed storage formats, which leads to non-affine loop nests and a vast, complex schedule space. The performance of a given schedule is sensitive to the sparsity pattern of the input tensors, making it difficult to find a single optimal solution. When input tensors in the same tensor contraction have conflicting data layouts in relation to the iteration order, it requires costly-both in time and memory-layout transformation, such as transposition. A promising but under-explored alternative is to generate a schedule that avoids explicit transposition, but this has not been systematically supported in existing compilers. This paper presents a new code generation strategy that generalizes the intermediate representation of the TACO sparse tensor compiler to generate a single loop nest, circumventing explicit transposition of tensors when the tensors have conflicting data layout iteration orders. We extend TACOś iteration graph to express a search-based strategy for locating elements in tensors with conflicting layouts, and we introduce new intermediate representation nodes to lower these schedules to efficient code. This enables the systematic generation of loops that do not require explicit transposition, thus avoiding the overhead of materializing temporary tensors. We evaluate our approach on a set of sparse tensor contractions using both real-world and synthetic datasets. Our results demonstrate that for computations with misaligned data layouts, our fused approach achieves up to 2x speedup for some sparsity patterns over the traditional approach of explicitly creating a transposed temporary. We also provide guidelines for when this new scheduling strategy is likely to be beneficial.

Tema degli esploratori: calcolo e infrastrutture (retro-catalogo arXiv: annunci del 2026-09-09 in cs.PL).
Categorie arXiv: Programming Languages (cs.PL).
Pagina del lavoro: https://arxiv.org/abs/2609.06328 — autori e testo completo sono lì (qui omessi i nomi: nessun dato personale nel deposito).
Licenza: titolo e abstract sono metadati arXiv distribuiti in CC0 1.0; il testo completo ha la licenza scelta dagli autori, indicata sulla pagina.
Letto il 2026-10-16T15:31:42.509Z dalla pagina https://arxiv.org/catchup/cs.PL/2026-09-09?abs=True (consentita dal robots.txt di arxiv.org, con il suo crawl-delay).