Repository navigation
Placement guidance for external structured quaternion kernel benchmark #48871
RQM-Technologies-dev
started this conversation in
Ideas
Replies: 1 comment
|
Update: the kernel has now completed whole-output N300 validation, a three-session one-device stability qualification, and profiler diagnostics. Following the current contribution path, I opened a narrowly scoped feature issue for the remaining placement and layout decision: #49887 No implementation PR will be opened until maintainers indicate whether this belongs as a TT-Metalium programming example or an experimental TT-NN |
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
Uh oh!
There was an error while loading. Please reload this page.
Hi Tenstorrent maintainers,
RQM Technologies has started an independent open-source project called
tt-rqm-kernels.The project defines CPU/PyTorch reference kernels and a benchmark suite called StructuredBench for structured tensor operators. The basic idea is simple: some things like rotation, phase, orientation, and other relationships are better represented with polar coordinates.
The first structure we are targeting is a quaternion-style four-lane value:
[N, 4] = [real, i, j, k]
In practical terms, this gives ordinary floating-point tensors a compact way to carry rotation, phase, orientation, and geometric state. That matters for workloads such as robotics pose updates, graphics rotation streams, wireless phase tracking, imaging pipelines, wave simulation, physical AI, scientific computing, and downstream defense applications.
The first proposed kernel is
qmul, the Hamilton product for two[N, 4]inputs producing one[N, 4]output. The reason this is useful as a benchmark is that it is small enough to validate, but more structured than a generic elementwise operation: every output lane depends on every input lane through a fixed multiply/add/sign pattern.Why this may be useful to Tenstorrent:
qmulexample, then possible TT-NN wrapping later.My question for you is:
Where should a minimal TT-Metalium
qmulexample for[N, 4]structured tensors live?qmulexample?StructuredBench currently reports throughput, latency, numerical error, estimated FLOPs/sec, effective GB/sec, arithmetic intensity, and scalar-reference spot-check error. The intent is to make future TT-Metalium or TT-NN results directly comparable against the CPU/PyTorch reference.
Repo:
RQM-Technologies-dev/tt-rqm-kernelsRelevant docs:
docs/structuredbench-spec.mddocs/operator-contracts.mddocs/tenstorrent-rfc.mdreports/tenstorrent_packet.mdAny guidance on preferred placement, benchmark format, or custom-op pathway would be appreciated.
All reactions