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