Skip to content

Migrate RoPE config to per layer#405

Merged
uuuvn merged 10 commits into
mainfrom
migrate_models_rope_config
May 15, 2026
Merged

Migrate RoPE config to per layer#405
uuuvn merged 10 commits into
mainfrom
migrate_models_rope_config

Conversation

@CC-Yeh
Copy link
Copy Markdown
Contributor

@CC-Yeh CC-Yeh commented May 13, 2026

  • Migrate RoPE to per layer config
  • Implement UnpackQK for non-RoPE Attention

@CC-Yeh CC-Yeh force-pushed the migrate_models_rope_config branch from 258fd11 to d92bdfd Compare May 13, 2026 13:01
@CC-Yeh CC-Yeh marked this pull request as ready for review May 13, 2026 13:59
Copy link
Copy Markdown

@chatgpt-codex-connector chatgpt-codex-connector Bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

💡 Codex Review

Here are some automated review suggestions for this pull request.

Reviewed commit: bce9ca4614

ℹ️ About Codex in GitHub

Your team has set up Codex to review pull requests in this repo. Reviews are triggered when you

  • Open a pull request for review
  • Mark a draft as ready
  • Comment "@codex review".

If Codex has suggestions, it will comment; otherwise it will react with 👍.

Codex can also answer questions or update the PR. Try commenting "@codex address that feedback".

Comment thread crates/backend-uzu/src/forward_pass/state/shared_buffers.rs
Comment thread crates/backend-uzu/src/config/decoder.rs Outdated
@CC-Yeh CC-Yeh force-pushed the migrate_models_rope_config branch from 14c71e1 to bce9ca4 Compare May 13, 2026 14:42
@uuuvn
Copy link
Copy Markdown
Contributor

uuuvn commented May 13, 2026

Check discord, lalamo 0.8.0 with your fix is in the release pipeline, once it's in the cdn we can just bump TOOLCHAIN_VERSION in this pr and everything should hopefully just work

@CC-Yeh CC-Yeh requested a review from uuuvn May 14, 2026 01:50
Comment thread crates/backend-uzu/src/backends/metal/kernel/attention/qk_unpack.metal Outdated
Comment thread crates/backend-uzu/src/backends/metal/kernel/attention/qk_unpack.metal Outdated
Comment thread crates/backend-uzu/src/config/decoder.rs Outdated
Comment thread crates/backend-uzu/src/config/generation.rs Outdated
Comment thread crates/backend-uzu/src/config/transformer_layer.rs Outdated
Comment thread crates/backend-uzu/src/config/transformer_layer.rs Outdated
Comment thread crates/backend-uzu/src/config/transformer_layer.rs Outdated
@CC-Yeh CC-Yeh force-pushed the migrate_models_rope_config branch 2 times, most recently from 4d432d0 to 6b795b7 Compare May 14, 2026 15:00
@CC-Yeh CC-Yeh requested a review from uuuvn May 14, 2026 15:02
@uuuvn uuuvn enabled auto-merge (squash) May 14, 2026 16:11
@CC-Yeh CC-Yeh force-pushed the migrate_models_rope_config branch from 9306107 to e2c77cf Compare May 15, 2026 03:02
@CC-Yeh CC-Yeh requested a review from alexxale as a code owner May 15, 2026 03:02
@uuuvn uuuvn merged commit 95b9ffe into main May 15, 2026
11 of 14 checks passed
@uuuvn uuuvn deleted the migrate_models_rope_config branch May 15, 2026 03:48
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.

2 participants