SIGN IN SIGN UP

aten.select.int (#3033)

Summary:
Pull Request resolved: https://github.com/pytorch/executorch/pull/3033

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.

Reviewed By: SS-JIA

Differential Revision: D56082483

fbshipit-source-id: f3a28712714034375eb86f6f5c6b6a3e23d525e8
J
Justin Yip committed
458d743bc86387fa4a3190f5d716e65584b41402
Parent: d0208d0
Committed by Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> on 4/16/2024, 4:05:19 AM