Skip to content

Add transpose, power, and where operations with tests#301

Open
zoecarver wants to merge 2 commits intozoecarver/reduce-opsfrom
zoecarver/transpose-power-where
Open

Add transpose, power, and where operations with tests#301
zoecarver wants to merge 2 commits intozoecarver/reduce-opsfrom
zoecarver/transpose-power-where

Conversation

@zoecarver
Copy link
Contributor

Adds three operations that round out the compute op coverage.

Transpose:

  • Follows the CB-input pattern (reads from CB, writes to DST)
  • Swaps tile width and height via TTKernel transpose intrinsic
  • Single-tile operation with straightforward 1:1 lowering

Power:

  • DST-based operation (not CB-input) - reads from DST, writes to DST
  • Takes integer exponent as an attribute
  • Lowering emits repeated multiplication or uses TTKernel power intrinsic depending on exponent

Where:

  • DST-based ternary operation for element-wise conditional selection
  • All three operands (condition, true_value, false_value) come from DST
  • Lowering emits TTKernel select/where intrinsic

Self-contained. Risk is limited to these new ops (adds new lowering, does not modify existing lowering patterns).

@zoecarver zoecarver requested a review from a team as a code owner February 2, 2026 16:18
@zoecarver zoecarver force-pushed the zoecarver/reduce-ops branch from bc85e71 to e74c621 Compare February 2, 2026 16:59
@zoecarver zoecarver force-pushed the zoecarver/transpose-power-where branch from 0cb029a to d0b0488 Compare February 2, 2026 16:59
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant