File tree Expand file tree Collapse file tree 2 files changed +15
-4
lines changed Expand file tree Collapse file tree 2 files changed +15
-4
lines changed Original file line number Diff line number Diff line change @@ -32,9 +32,9 @@ test_buck2_select_ops_in_list() {
32
32
${PYTHON_EXECUTABLE} -m examples.export.export_example --model_name=" add_mul"
33
33
34
34
echo " Running selective build test"
35
- # set max_kernel_num=16: 13 primops, add, mul
35
+ # set max_kernel_num=17: 14 primops, add, mul
36
36
$BUCK run //examples/selective_build:selective_build_test \
37
- --config=executorch.max_kernel_num=16 \
37
+ --config=executorch.max_kernel_num=17 \
38
38
--config=executorch.select_ops=list -- --model_path=./add_mul.pte
39
39
40
40
echo " Removing add_mul.pte"
@@ -80,12 +80,12 @@ test_cmake_select_ops_in_list() {
80
80
echo " Exporting MobilenetV2"
81
81
${PYTHON_EXECUTABLE} -m examples.export.export_example --model_name=" mv2"
82
82
83
- # set MAX_KERNEL_NUM=16: 13 primops, add, mul
83
+ # set MAX_KERNEL_NUM=17: 14 primops, add, mul
84
84
(rm -rf cmake-out \
85
85
&& mkdir cmake-out \
86
86
&& cd cmake-out \
87
87
&& retry cmake -DBUCK2=" $BUCK " \
88
- -DMAX_KERNEL_NUM=16 \
88
+ -DMAX_KERNEL_NUM=17 \
89
89
-DBUILD_SELECTIVE_BUILD_TEST=ON \
90
90
-DCMAKE_BUILD_TYPE=Release \
91
91
-DSELECT_OPS_LIST=" aten::convolution.out,\
Original file line number Diff line number Diff line change @@ -210,6 +210,17 @@ static Kernel prim_ops[] = {
210
210
BOOLEAN_ET_PRIM_OP (<=, stack, context);
211
211
}),
212
212
213
+ // executorch_prim::floordiv.int(int, int) -> int
214
+ Kernel (
215
+ " executorch_prim::floordiv.int" ,
216
+ [](RuntimeContext& context, EValue** stack) {
217
+ (void )context;
218
+ EValue& a = *stack[0 ];
219
+ EValue& b = *stack[1 ];
220
+ EValue& out = *stack[2 ];
221
+ out = EValue (a.toInt () / b.toInt ());
222
+ }),
223
+
213
224
// executorch_prim::et_copy_index.tensor(tensor, tensor) -> tensor
214
225
Kernel (" executorch_prim::et_copy_index.tensor" , &et_copy_index),
215
226
You can’t perform that action at this time.
0 commit comments