Skip to content

Commit bc176aa

Browse files
JacobSzwejbkafacebook-github-bot
authored andcommitted
bump num kernels (#421)
Summary: Pull Request resolved: #421 we added another prim op Reviewed By: lucylq Differential Revision: D49467043 fbshipit-source-id: 57b31966b9f61f4a7775a1247feb424e5ef15f28
1 parent a216a03 commit bc176aa

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

examples/selective_build/test_selective_build.sh

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -32,9 +32,9 @@ test_buck2_select_ops_in_list() {
3232
${PYTHON_EXECUTABLE} -m examples.export.export_example --model_name="add_mul"
3333

3434
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
3636
$BUCK run //examples/selective_build:selective_build_test \
37-
--config=executorch.max_kernel_num=16 \
37+
--config=executorch.max_kernel_num=17 \
3838
--config=executorch.select_ops=list -- --model_path=./add_mul.pte
3939

4040
echo "Removing add_mul.pte"
@@ -80,12 +80,12 @@ test_cmake_select_ops_in_list() {
8080
echo "Exporting MobilenetV2"
8181
${PYTHON_EXECUTABLE} -m examples.export.export_example --model_name="mv2"
8282

83-
# set MAX_KERNEL_NUM=16: 13 primops, add, mul
83+
# set MAX_KERNEL_NUM=17: 14 primops, add, mul
8484
(rm -rf cmake-out \
8585
&& mkdir cmake-out \
8686
&& cd cmake-out \
8787
&& retry cmake -DBUCK2="$BUCK" \
88-
-DMAX_KERNEL_NUM=16 \
88+
-DMAX_KERNEL_NUM=17 \
8989
-DBUILD_SELECTIVE_BUILD_TEST=ON \
9090
-DCMAKE_BUILD_TYPE=Release \
9191
-DSELECT_OPS_LIST="aten::convolution.out,\

0 commit comments

Comments
 (0)