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

[js/webgpu] Optimize Expand #22752

Merged
merged 1 commit into from
Nov 12, 2024
Merged

[js/webgpu] Optimize Expand #22752

merged 1 commit into from
Nov 12, 2024

Conversation

qjia7
Copy link
Contributor

@qjia7 qjia7 commented Nov 7, 2024

Use components = 4 if possible.

llama3.2-1B becomes 20 tokens/s from 18 tokens/s on my iGPUs.

Use components = 4 if possible.

llama3.2-1B becomes 20 tokens/s from 18 tokens/s on my iGPUs.
@qjia7
Copy link
Contributor Author

qjia7 commented Nov 7, 2024

@guschmue @fs-eire Please take a look, thanks.

@guschmue
Copy link
Contributor

/azp run ONNX Runtime Web CI Pipeline,Windows GPU CI Pipeline,Linux Android Emulator QNN CI Pipeline

@guschmue
Copy link
Contributor

/azp run Linux CPU CI Pipeline,Linux CPU Minimal Build E2E CI Pipeline,Linux GPU CI Pipeline,Linux GPU TensorRT CI Pipeline,Linux OpenVINO CI Pipeline,Linux QNN CI Pipeline,MacOS CI Pipeline,Windows ARM64 QNN CI Pipeline,Windows CPU CI Pipeline

@guschmue
Copy link
Contributor

/azp run Windows GPU TensorRT CI Pipeline,onnxruntime-binary-size-checks-ci-pipeline,orttraining-linux-ci-pipeline,orttraining-linux-gpu-ci-pipeline,orttraining-ortmodule-distributed,Windows x64 QNN CI Pipeline,Big Models

Copy link

Azure Pipelines successfully started running 1 pipeline(s).

Copy link

Azure Pipelines could not run because the pipeline triggers exclude this branch/path.

Copy link

Azure Pipelines successfully started running 1 pipeline(s).

@guschmue
Copy link
Contributor

/azp run Windows GPU CUDA CI Pipeline,Windows GPU DML CI Pipeline,Windows GPU Doc Gen CI Pipeline

Copy link

Azure Pipelines could not run because the pipeline triggers exclude this branch/path.

@guschmue guschmue added the ep:WebGPU ort-web webgpu provider label Nov 12, 2024
@guschmue guschmue merged commit 7e0dd9d into microsoft:main Nov 12, 2024
50 checks passed
ishwar-raut1 pushed a commit to ishwar-raut1/onnxruntime that referenced this pull request Nov 19, 2024
Use components = 4 if possible.

llama3.2-1B becomes 20 tokens/s from 18 tokens/s on my iGPUs.
guschmue pushed a commit that referenced this pull request Dec 2, 2024
Use components = 4 if possible.

llama3.2-1B becomes 20 tokens/s from 18 tokens/s on my iGPUs.
@qjia7 qjia7 deleted the opt_expand branch December 9, 2024 05:00
@qjia7 qjia7 mentioned this pull request Dec 9, 2024
guschmue pushed a commit that referenced this pull request Dec 10, 2024
### Description
<!-- Describe your changes. -->
Use components = 4 if possible.

This is the webgpu native implementation from #22752
ankitm3k pushed a commit to intel/onnxruntime that referenced this pull request Dec 11, 2024
Use components = 4 if possible.

llama3.2-1B becomes 20 tokens/s from 18 tokens/s on my iGPUs.
ankitm3k pushed a commit to intel/onnxruntime that referenced this pull request Dec 11, 2024
### Description
<!-- Describe your changes. -->
Use components = 4 if possible.

This is the webgpu native implementation from microsoft#22752
ankitm3k pushed a commit to intel/onnxruntime that referenced this pull request Dec 11, 2024
Use components = 4 if possible.

llama3.2-1B becomes 20 tokens/s from 18 tokens/s on my iGPUs.
ankitm3k pushed a commit to intel/onnxruntime that referenced this pull request Dec 11, 2024
### Description
<!-- Describe your changes. -->
Use components = 4 if possible.

This is the webgpu native implementation from microsoft#22752
ankitm3k pushed a commit to intel/onnxruntime that referenced this pull request Dec 11, 2024
### Description
<!-- Describe your changes. -->
Use components = 4 if possible.

This is the webgpu native implementation from microsoft#22752
tarekziade pushed a commit to tarekziade/onnxruntime that referenced this pull request Jan 10, 2025
### Description
<!-- Describe your changes. -->
Use components = 4 if possible.

This is the webgpu native implementation from microsoft#22752
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
ep:WebGPU ort-web webgpu provider
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants