aten.select.int

[yipjustin]

Summary:
Port over the select.int shaders to ET.

1. Since in ET, tensor-shape reasoning happens in AOT, therefore we can simplify the c++ caller code by a lot.
2. In this diff, we also try to use the same buffer object for passing arguments to all shaders. Not worry about perf cost, since cost difference between passing int and ivec4 is very minor.

Differential Revision: D56082483

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

❌ 6 New Failures

As of commit e0db27e with merge base d0208d0 ():

NEW FAILURES - The following jobs have failed:

• pull / test-llama-runner-linux (fp32, buck2, portable) / linux-job (gh)
  RuntimeError: Command docker exec -t c3590d37aed375a5413e5336ab3c322217370bda43a6e3b4037dcdb82fc57386 /exec failed with exit code 3
• pull / test-llama-runner-linux (fp32, buck2, xnnpack+kv+custom) / linux-job (gh)
  RuntimeError: Command docker exec -t 5ead4612f21cc60293e9acc70190cc42f0dd383937b710d16fd64d0af1ea1904 /exec failed with exit code 3
• pull / test-llama-runner-linux (fp32, cmake, portable) / linux-job (gh)
  RuntimeError: Command docker exec -t 11572132ae204ced83f040dc9525e44c5d0333ba825d0f41df3175e5cae7b106 /exec failed with exit code 2
• pull / test-llama-runner-linux (fp32, cmake, xnnpack+kv+custom) / linux-job (gh)
  RuntimeError: Command docker exec -t c855813d76600aae504f5657c88470a80b42a2714058baa6ccb648c985e1446e /exec failed with exit code 2
• pull / test-llama-runner-linux-android (cmake) / linux-job (gh)
  RuntimeError: Command docker exec -t e833be1ef8ba3a91686d72d6171a6d9600010d9d2de9c97f0f3667291ad9210e /exec failed with exit code 2
• pull / unittest / linux (buck2) / linux-job (gh)
  RuntimeError: Command docker exec -t 2b1fa4d9f799a2127571cae9e14494ef49ffce07090ba7764b01770ba47fb6a0 /exec failed with exit code 1

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

[facebook-github-bot]

This pull request was exported from Phabricator. Differential Revision: D56082483

[facebook-github-bot]

This pull request was exported from Phabricator. Differential Revision: D56082483

[facebook-github-bot]

This pull request was exported from Phabricator. Differential Revision: D56082483

[facebook-github-bot]

This pull request has been merged in 458d743.