Skip to content

[WebNN EP] Add outputDataType option for the ArgMax/ArgMin ops#21385

Merged
fdwr merged 1 commit into
microsoft:mainfrom
Honry:argmaxmin-outputDataType
Jul 22, 2024
Merged

[WebNN EP] Add outputDataType option for the ArgMax/ArgMin ops#21385
fdwr merged 1 commit into
microsoft:mainfrom
Honry:argmaxmin-outputDataType

Conversation

@Honry

@Honry Honry commented Jul 17, 2024

Copy link
Copy Markdown
Contributor

Description

WebNN spec introduces a new option: outputDataType to argMax and argMin ops, it's default value is int32, we should explicitly set it to int64 for WebNN EP.

Spec CR: "Add outputDataType to argmin/argmax" webmachinelearning/webnn#730

@Honry

Honry commented Jul 17, 2024

Copy link
Copy Markdown
Contributor Author

@fdwr, @guschmue, PTAL, thanks!

cc/ @huningxin

fdwr
fdwr previously approved these changes Jul 17, 2024
@fdwr

fdwr commented Jul 17, 2024

Copy link
Copy Markdown
Contributor

@Honry Thank you for updating it promptly to match the Chromium implementation.

@fdwr

fdwr commented Jul 17, 2024

Copy link
Copy Markdown
Contributor

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

@fdwr

fdwr commented Jul 17, 2024

Copy link
Copy Markdown
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

@fdwr

fdwr commented Jul 17, 2024

Copy link
Copy Markdown
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

@azure-pipelines

Copy link
Copy Markdown
Azure Pipelines successfully started running 3 pipeline(s).

@azure-pipelines

Copy link
Copy Markdown
Azure Pipelines successfully started running 9 pipeline(s).

@azure-pipelines

Copy link
Copy Markdown
Azure Pipelines successfully started running 7 pipeline(s).

@fdwr

fdwr commented Jul 18, 2024

Copy link
Copy Markdown
Contributor

/azp run Windows GPU CI Pipeline

@azure-pipelines

Copy link
Copy Markdown
Azure Pipelines successfully started running 1 pipeline(s).

@huningxin huningxin left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

LGTM with a comment

@fdwr

fdwr commented Jul 18, 2024

Copy link
Copy Markdown
Contributor

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

@fdwr

fdwr commented Jul 18, 2024

Copy link
Copy Markdown
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

@fdwr

fdwr commented Jul 18, 2024

Copy link
Copy Markdown
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

@azure-pipelines

Copy link
Copy Markdown
Azure Pipelines successfully started running 3 pipeline(s).

@azure-pipelines

Copy link
Copy Markdown
Azure Pipelines successfully started running 7 pipeline(s).

@azure-pipelines

Copy link
Copy Markdown
Azure Pipelines successfully started running 9 pipeline(s).

@fdwr

fdwr commented Jul 19, 2024

Copy link
Copy Markdown
Contributor

/azp run Windows GPU TensorRT CI Pipeline

@fdwr

fdwr commented Jul 19, 2024

Copy link
Copy Markdown
Contributor

/azp run Windows CPU CI Pipeline, Windows GPU CI Pipeline

fdwr
fdwr previously approved these changes Jul 19, 2024
@fdwr

fdwr commented Jul 19, 2024

Copy link
Copy Markdown
Contributor

@Honry I don't know why the GPU tests keep getting canceled. Maybe there is a change in main we should merge into this branch?

@Honry

Honry commented Jul 19, 2024

Copy link
Copy Markdown
Contributor Author

@Honry I don't know why the GPU tests keep getting canceled. Maybe there is a change in main we should merge into this branch?

I will do a rebase.

@Honry Honry force-pushed the argmaxmin-outputDataType branch from 6920447 to ca89be0 Compare July 19, 2024 02:34
@Honry

Honry commented Jul 19, 2024

Copy link
Copy Markdown
Contributor Author

@Honry I don't know why the GPU tests keep getting canceled. Maybe there is a change in main we should merge into this branch?

I will do a rebase.

@fdwr, done, please help retrigger the CI.

@fdwr

fdwr commented Jul 19, 2024

Copy link
Copy Markdown
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

@fdwr

fdwr commented Jul 19, 2024

Copy link
Copy Markdown
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

@fdwr

fdwr commented Jul 19, 2024

Copy link
Copy Markdown
Contributor

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

1 similar comment
@guschmue

Copy link
Copy Markdown
Contributor

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

@guschmue

Copy link
Copy Markdown
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
Copy Markdown
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

@azure-pipelines

Copy link
Copy Markdown
Azure Pipelines successfully started running 3 pipeline(s).

@azure-pipelines

Copy link
Copy Markdown
Azure Pipelines successfully started running 9 pipeline(s).

@azure-pipelines

Copy link
Copy Markdown
Azure Pipelines successfully started running 7 pipeline(s).

@fdwr

fdwr commented Jul 20, 2024

Copy link
Copy Markdown
Contributor

/azp run Windows CPU CI Pipeline, Windows GPU CI Pipeline

@azure-pipelines

Copy link
Copy Markdown
Azure Pipelines successfully started running 2 pipeline(s).

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ep:WebNN WebNN execution provider

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants