Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[sharktank] Revert "[llama] Added the fused rotary embedding kernel (#719)" #752

Merged
merged 5 commits into from
Jan 10, 2025

Conversation

archana-ramalingam
Copy link
Collaborator

@archana-ramalingam archana-ramalingam commented Jan 4, 2025

This reverts commit fc9576b, which causes segmentation fault at Llama prefill in IREE runtime.

@archana-ramalingam archana-ramalingam changed the title Revert "[llama] Added the fused rotary embedding kernel (#719)" [sharktank] Revert "[llama] Added the fused rotary embedding kernel (#719)" Jan 4, 2025
archana-ramalingam added a commit that referenced this pull request Jan 8, 2025
Temporarily disable IREE perplexity tests till #752 is investigated and
merged.

---------

Co-authored-by: Scott Todd <[email protected]>
@archana-ramalingam archana-ramalingam merged commit 63ff841 into main Jan 10, 2025
33 checks passed
@archana-ramalingam archana-ramalingam deleted the update-ppl-pre-submit branch January 10, 2025 20:58
stbaione added a commit to stbaione/SHARK-Platform that referenced this pull request Jan 15, 2025
stbaione added a commit that referenced this pull request Jan 16, 2025
…kernel (#719)" (#752)" (#828)

This reverts commit 63ff841.

# Description

Our [user
docs](https://github.com/nod-ai/shark-ai/blob/main/docs/shortfin/llm/user/llama_serving.md)
started producing repetitive output:

```bash
curl http://localhost:8003/generate \
    -H "Content-Type: application/json" \
    -d '{
        "text": "Name the capital of the United States.",
        "sampling_params": {"max_completion_tokens": 50}
    }'

data:  Washington, D.C.! Washington, D.C.! Washington, D.C.! Washington, D.C.! Washington, D.C.! Washington, D.C.! Washington, D.C.! Washington, D.C.! Washington, D.C.! Washington, D.C.! Washington
```

After bisecting, the problematic commit was found to be
63ff841.

Without commit reverted:

```bash
curl http://localhost:8003/generate \
    -H "Content-Type: application/json" \
    -d '{
        "text": "Name the capital of the United States.",
        "sampling_params": {"max_completion_tokens": 50}
    }'

data:  Washington, D.C.! Washington, D.C.! Washington, D.C.! Washington, D.C.! Washington, D.C.! Washington, D.C.! Washington, D.C.! Washington, D.C.! Washington, D.C.! Washington, D.C.! Washington
```

With commit reverted:

```bash
curl http://localhost:8003/generate \
    -H "Content-Type: application/json" \
    -d '{
        "text": "Name the capital of the United States.",
        "sampling_params": {"max_completion_tokens": 50}
    }'

data:  Washington D.C. is the capital of the United States. Washington D.C. is a special district that is not part of any state but is the capital of the United States. Washington D.C. is a federal district that serves as the seat of the
```
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