GitHub - oakzeng/CuTe: Helper function to calculate tensor elements' offset, where tensor is represented by Cute layout defined in https://arxiv.org/pdf/2603.02298

Branches Tags

Name		Name	Last commit message	Last commit date
Latest commit History 3 Commits
LICENSE		LICENSE
Makefile		Makefile
README		README
cute_layout.hpp		cute_layout.hpp
test_cute.cpp		test_cute.cpp
test_pim.cpp		test_pim.cpp
test_slc.cpp		test_slc.cpp

Repository files navigation

Implement a helper function cute_offset to calculate tensor element's
offset in the unit of element (assuming the first tensor element at
offset 0).

A tensor layout is represented as (shape : stride) layout. See more details
in paper https://arxiv.org/pdf/2603.02298