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

Add simplified layernorm fusion for Gemma #20572

Merged

Conversation

PatriceVignola
Copy link
Contributor

Gemma has a Mul node right after the Gather and before the first layer norm.

@PatriceVignola
Copy link
Contributor Author

/azp run Windows CPU CI Pipeline

@PatriceVignola
Copy link
Contributor Author

/azp run orttraining-linux-gpu-ci-pipeline

Copy link

Azure Pipelines successfully started running 1 pipeline(s).

@PatriceVignola
Copy link
Contributor Author

/azp run orttraining-ortmodule-distributed

Copy link

Azure Pipelines successfully started running 1 pipeline(s).

1 similar comment
Copy link

Azure Pipelines successfully started running 1 pipeline(s).

@PatriceVignola
Copy link
Contributor Author

/azp run Windows CPU CI Pipeline
/azp run orttraining-linux-gpu-ci-pipeline
/azp run orttraining-ortmodule-distributed

Copy link

No pipelines are associated with this pull request.

@PatriceVignola
Copy link
Contributor Author

/azp run Windows CPU CI Pipeline

@PatriceVignola
Copy link
Contributor Author

/azp run orttraining-linux-gpu-ci-pipeline

Copy link

Azure Pipelines successfully started running 1 pipeline(s).

@PatriceVignola
Copy link
Contributor Author

/azp run orttraining-ortmodule-distributed

Copy link

Azure Pipelines successfully started running 1 pipeline(s).

1 similar comment
Copy link

Azure Pipelines successfully started running 1 pipeline(s).

@PatriceVignola
Copy link
Contributor Author

/azp run orttraining-linux-gpu-ci-pipeline

@PatriceVignola
Copy link
Contributor Author

/azp run orttraining-ortmodule-distributed

Copy link

Azure Pipelines successfully started running 1 pipeline(s).

1 similar comment
Copy link

Azure Pipelines successfully started running 1 pipeline(s).

@PatriceVignola
Copy link
Contributor Author

/azp run orttraining-linux-gpu-ci-pipeline

@PatriceVignola
Copy link
Contributor Author

/azp run orttraining-ortmodule-distributed

Copy link

Azure Pipelines successfully started running 1 pipeline(s).

1 similar comment
Copy link

Azure Pipelines successfully started running 1 pipeline(s).

@PatriceVignola PatriceVignola merged commit 478d3e0 into main May 7, 2024
87 of 93 checks passed
@PatriceVignola PatriceVignola deleted the user/pavignol/add-gemma-simplified-layernorm-fusion branch May 7, 2024 03:07
TedThemistokleous pushed a commit to TedThemistokleous/onnxruntime that referenced this pull request May 7, 2024
Gemma has a `Mul` node right after the `Gather` and before the first
layer norm.
poweiw pushed a commit to poweiw/onnxruntime that referenced this pull request Jun 25, 2024
Gemma has a `Mul` node right after the `Gather` and before the first
layer norm.
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