Skip to content

Commit 49928bc

Browse files
yipjustinfacebook-github-bot
authored andcommitted
select_copy.int (#3085)
Summary: Pull Request resolved: #3085 equivalent to `select.int` ghstack-source-id: 222407935 exported-using-ghexport Reviewed By: SS-JIA Differential Revision: D56092143 fbshipit-source-id: 2959069d87cef6f08aa0960e2f10a9416eb4109d
1 parent 78cb141 commit 49928bc

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

backends/vulkan/runtime/graph/ops/impl/Select.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -127,6 +127,7 @@ void select_int(ComputeGraph& graph, const std::vector<ValueRef>& args) {
127127

128128
REGISTER_OPERATORS {
129129
VK_REGISTER_OP(aten.select.int, select_int);
130+
VK_REGISTER_OP(aten.select_copy.int, select_int);
130131
}
131132

132133
} // namespace vkcompute

backends/vulkan/test/op_tests/cases.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -182,4 +182,5 @@ def get_select_int_inputs():
182182
"aten.native_layer_norm.default": get_native_layer_norm_inputs(),
183183
"aten.full.default": get_full_inputs(),
184184
"aten.select.int": get_select_int_inputs(),
185+
"aten.select_copy.int": get_select_int_inputs(),
185186
}

0 commit comments

Comments
 (0)