The Python PyCuTe mirror is used by the backend code generator to reason about tensor layouts, but its…
The Python PyCuTe mirror is used by the backend code generator to reason about tensor layouts, but its…: a task in MiMo-V2.6-RL-harbor-code: MiMo-V2.6-RL Code (Harbor) (Harbor dataset). A Swizzle(bits, base, shift) must expose the same layout-like extent API as Layout: size() is 2 (bits + base +…
The task
A Swizzle(bits, base, shift) must expose the same layout-like extent API as Layout: size() is 2 ** (bits + base + abs(shift)) and cosize() is the same extent. The zero-bit case (bits=0, base=3, shift=0) is valid and maps every offset to itself; positive and negative shifts must both remain bijections over their…
Part of FineEnvs/MiMo-V2.6-RL-harbor-code.