onnxruntime icon indicating copy to clipboard operation
onnxruntime copied to clipboard

[WIP] Contrib operator for Deformable Multi-Scale Attention

Open duanqn opened this issue 1 year ago • 4 comments

Description

Introduce a contrib operator for Deformable Multi-Scale Attention

Motivation and Context

Deformable Multi-Scale Attention is first proposed in Deformable DETR. Implementing it with official ONNX operators results in poor performance on CPU. We implemented with AVX-512 intrinsics and the performance improved ~20x with the same input size. Note that our optimized implementation only works for certain parameter values.

duanqn avatar Apr 03 '24 04:04 duanqn

/azp run Windows ARM64 QNN CI Pipeline,Windows x64 QNN CI Pipeline,Windows CPU CI Pipeline,Windows GPU CI Pipeline,Windows GPU TensorRT CI Pipeline,ONNX Runtime Web CI Pipeline,Linux CPU CI Pipeline,Linux CPU Minimal Build E2E CI Pipeline,Linux GPU CI Pipeline,Linux GPU TensorRT CI Pipeline

tianleiwu avatar Apr 05 '24 04:04 tianleiwu

/azp run Linux OpenVINO CI Pipeline,Linux QNN CI Pipeline,MacOS CI Pipeline,orttraining-amd-gpu-ci-pipeline,orttraining-linux-ci-pipeline,orttraining-linux-gpu-ci-pipeline,orttraining-ortmodule-distributed,onnxruntime-binary-size-checks-ci-pipeline,Big Models

tianleiwu avatar Apr 05 '24 04:04 tianleiwu

Azure Pipelines successfully started running 8 pipeline(s).

azure-pipelines[bot] avatar Apr 05 '24 04:04 azure-pipelines[bot]

Azure Pipelines successfully started running 10 pipeline(s).

azure-pipelines[bot] avatar Apr 05 '24 04:04 azure-pipelines[bot]

/azp run Windows ARM64 QNN CI Pipeline,Windows x64 QNN CI Pipeline,Windows CPU CI Pipeline,Windows GPU CI Pipeline,Windows GPU TensorRT CI Pipeline,ONNX Runtime Web CI Pipeline,Linux CPU CI Pipeline,Linux CPU Minimal Build E2E CI Pipeline,Linux GPU CI Pipeline,Linux GPU TensorRT CI Pipeline

tianleiwu avatar Aug 05 '24 18:08 tianleiwu

/azp run Linux OpenVINO CI Pipeline,Linux QNN CI Pipeline,MacOS CI Pipeline,orttraining-amd-gpu-ci-pipeline,orttraining-linux-ci-pipeline,orttraining-linux-gpu-ci-pipeline,orttraining-ortmodule-distributed,onnxruntime-binary-size-checks-ci-pipeline,Big Models

tianleiwu avatar Aug 05 '24 18:08 tianleiwu

Azure Pipelines successfully started running 9 pipeline(s).

azure-pipelines[bot] avatar Aug 05 '24 18:08 azure-pipelines[bot]

Azure Pipelines successfully started running 9 pipeline(s).

azure-pipelines[bot] avatar Aug 05 '24 18:08 azure-pipelines[bot]