Skip to content

add RMSNorm kernel to roofline plots in the latency-demo notebook #165

Description

@yuhaohaoyu

Lineage and Dependency

what's the goal

Add synthetic latency experiments in to the roofline sections of the latency_demo notebook.

What's the benefit

RMSNorm is an important normalization kernel used in IBM's Granite, Llama-3, Gemma, etc. Including it in ktir-cpu's latency demo notebook is great addition to showcase the usefulness of ktir-cpu.

how to assess completion

  1. Update codes under notebooks/, with codes to parameterization of the kernel to full fill the 4-core, 32-core-strong-scaling, and 32-core-weak-scaling setup for the roofline section of latency_demo notebook.
  2. Notebook run output (capture into html) for visual verification.

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