Qualcomm AI Engine Direct - Support min.dim, max.dim and argmax ops

[DannyYuyang-quic]

Summary:

• Support aten.min.dim, aten.max.dim and aten.argmax

Test plan

aten.min.dim

python backends/qualcomm/tests/test_qnn_delegate.py TestQNNQuantizedOperators.test_qnn_backend_min_dim -s ${device_id} -H ${host_id} -m ${soc} -b build-android

python backends/qualcomm/tests/test_qnn_delegate.py TestQNNFloatingPointOperator.test_qnn_backend_min_dim -s ${device_id} -H ${host_id} -m ${soc} -b build-android

aten.max.dim

python backends/qualcomm/tests/test_qnn_delegate.py TestQNNQuantizedOperators.test_qnn_backend_max_dim -s ${device_id} -H ${host_id} -m ${soc} -b build-android

python backends/qualcomm/tests/test_qnn_delegate.py TestQNNFloatingPointOperator.test_qnn_backend_max_dim -s ${device_id} -H ${host_id} -m ${soc} -b build-android

argmax

python backends/qualcomm/tests/test_qnn_delegate.py TestQNNQuantizedOperators.test_qnn_backend_argmax -s ${device_id} -H ${host_id} -m ${soc} -b build-android

python backends/qualcomm/tests/test_qnn_delegate.py TestQNNFloatingPointOperator.test_qnn_backend_argmax -s ${device_id} -H ${host_id} -m ${soc} -b build-android

cc: @haowhsu-quic @cccclai

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/12669

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

❗ 1 Active SEVs

There are 1 currently active SEVs. If your PR is affected, please view them below:

• linux.aws.h100.8 instance is down, potentially longer queue on linux.aws.h100

❌ 1 New Failure

As of commit 5429220 with merge base e31eb56 ():

NEW FAILURE - The following job has failed:

• pull / test-eval_llama-mmlu-linux / linux-job (gh)
  RuntimeError: Command docker exec -t 2186b49a063b2184c2260c76b46ab0913ed8439fb3688928c2e747da2d3b72b9 /exec failed with exit code 1

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[DannyYuyang-quic]

Hi @cccclai ,
This PR covers part of the ops support requested in #12623
Please have a look
Thanks!

[DannyYuyang-quic]

@pytorchbot label "release notes: qualcomm"

[facebook-github-bot]

@cccclai has imported this pull request. If you are a Meta employee, you can view this in D78675516.

[cccclai]

Is the PR ready? It seems good to me but it's this pr is in draft

[cccclai]

Thanks for quickly adding the ops!

[DannyYuyang-quic]

Is the PR ready? It seems good to me but it's this pr is in draft

Yes, it's ready! I also added a few more comments in the pass to help with readability.

[facebook-github-bot]

@cccclai has imported this pull request. If you are a Meta employee, you can view this in D78675516.