Skip to content

create rope_fwd_ktir.mlir #166

Description

@yuhaohaoyu

Lineage and Dependency

what's the goal

To create a static rope_fwd_ktir.mlir example input file.

What's the benefit

RoPE is an important accuracy boosting kernel used in IBM's Granite, and nearly every since DeepSeek. Including it in ktir-cpu's latency demo notebook is a great addition to showcase the usefulness of ktir-cpu.

how to assess completion

  1. create examples/latency/rope_fwd_ktir.mlir program with reasonable problem size and grid size.
  2. tests to conform with representative rope operation, accuracy, and loads of datamove and compute.

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

Labels

documentationImprovements or additions to documentation

Projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions