[ET-VK][2/n] select_copy.int

yipjustin · yipjustin · commit d46653bb1e2c · 2024-04-16T23:33:36.000-07:00
equivalent to `select.int` Differential Revision: [D56092143](https://our.internmc.facebook.com/intern/diff/D56092143/) ghstack-source-id: 222407935 Pull Request resolved: #3085
diff --git a/backends/vulkan/runtime/graph/ops/impl/Select.cpp b/backends/vulkan/runtime/graph/ops/impl/Select.cpp
@@ -127,6 +127,7 @@ void select_int(ComputeGraph& graph, const std::vector<ValueRef>& args) {
 
 REGISTER_OPERATORS {
   VK_REGISTER_OP(aten.select.int, select_int);
+  VK_REGISTER_OP(aten.select_copy.int, select_int);
 }
 
 } // namespace vkcompute
diff --git a/backends/vulkan/test/op_tests/cases.py b/backends/vulkan/test/op_tests/cases.py
@@ -182,4 +182,5 @@ def get_select_int_inputs():
     "aten.native_layer_norm.default": get_native_layer_norm_inputs(),
     "aten.full.default": get_full_inputs(),
     "aten.select.int": get_select_int_inputs(),
+    "aten.select_copy.int": get_select_int_inputs(),
 }

Original file line number	Diff line number	Diff line change
`@@ -127,6 +127,7 @@ void select_int(ComputeGraph& graph, const std::vector<ValueRef>& args) {`
`127`	`127`
`128`	`128`	`REGISTER_OPERATORS {`
`129`	`129`	`VK_REGISTER_OP(aten.select.int, select_int);`
	`130`	`+ VK_REGISTER_OP(aten.select_copy.int, select_int);`
`130`	`131`	`}`
`131`	`132`
`132`	`133`	`} // namespace vkcompute`
Original file line number	Diff line number	Diff line change
`@@ -182,4 +182,5 @@ def get_select_int_inputs():`
`182`	`182`	`"aten.native_layer_norm.default": get_native_layer_norm_inputs(),`
`183`	`183`	`"aten.full.default": get_full_inputs(),`
`184`	`184`	`"aten.select.int": get_select_int_inputs(),`
	`185`	`+ "aten.select_copy.int": get_select_int_inputs(),`
`185`	`186`	`}`