[do not land] selective build size test

Author: lucylq

examples/selective_build/test_selective_build.sh

@@ -110,32 +110,29 @@ test_cmake_select_all_ops() {
 }
 
 test_cmake_select_ops_in_list() {
-    echo "Exporting MobilenetV2"
-    ${PYTHON_EXECUTABLE} -m examples.portable.scripts.export --model_name="mv2"
+    # echo "Exporting MobilenetV2"
+    # ${PYTHON_EXECUTABLE} -m examples.portable.scripts.export --model_name="mv2"
 
     local example_dir=examples/selective_build
     local build_dir=cmake-out/${example_dir}
     # set MAX_KERNEL_NUM=22: 19 primops, add, mul
     rm -rf ${build_dir}
     retry cmake -DCMAKE_BUILD_TYPE=Release \
-            -DMAX_KERNEL_NUM=22 \
-            -DEXECUTORCH_SELECT_OPS_LIST="aten::convolution.out,\
-aten::_native_batch_norm_legit_no_training.out,aten::hardtanh.out,aten::add.out,\
-aten::mean.out,aten::view_copy.out,aten::permute_copy.out,aten::addmm.out,\
-aten,aten::clone.out" \
+            -DEXECUTORCH_SELECT_OPS_LIST="aten::add.out" \
             -DCMAKE_INSTALL_PREFIX=cmake-out \
+            -DEXECUTORCH_OPTIMIZE_SIZE=ON \
             -DPYTHON_EXECUTABLE="$PYTHON_EXECUTABLE" \
             -B${build_dir} \
             ${example_dir}
 
     echo "Building ${example_dir}"
     cmake --build ${build_dir} -j9 --config Release
 
-    echo 'Running selective build test'
-    ${build_dir}/selective_build_test --model_path="./mv2.pte"
+    # echo 'Running selective build test'
+    # ${build_dir}/selective_build_test --model_path="./mv2.pte"
 
-    echo "Removing mv2.pte"
-    rm "./mv2.pte"
+    # echo "Removing mv2.pte"
+    # rm "./mv2.pte"
 }
 
 test_cmake_select_ops_in_yaml() {
@@ -168,15 +165,15 @@ fi
 
 if [[ -z $PYTHON_EXECUTABLE ]];
 then
-  PYTHON_EXECUTABLE=python3
+  PYTHON_EXECUTABLE=python
 fi
 
 if [[ $1 == "cmake" ]];
 then
     cmake_install_executorch_lib
-    test_cmake_select_all_ops
+    # test_cmake_select_all_ops
     test_cmake_select_ops_in_list
-    test_cmake_select_ops_in_yaml
+    # test_cmake_select_ops_in_yaml
 elif [[ $1 == "buck2" ]];
 then
     test_buck2_select_all_ops

kernels/portable/cpu/selective_build.h

@@ -16,12 +16,29 @@
 #include "selected_op_variants.h"
 #else
 // dummy implementation
+// inline constexpr bool should_include_kernel_dtype(
+//     const char* /*operator_name*/,
+//     executorch::aten::ScalarType /*scalar_type*/
+// ) {
+//   return true;
+// }
+
+// Only add kernel.
 inline constexpr bool should_include_kernel_dtype(
-    const char* /*operator_name*/,
-    executorch::aten::ScalarType /*scalar_type*/
-) {
-  return true;
+    const char* operator_name,
+    executorch::aten::ScalarType scalar_type) {
+  return ((std::string_view(operator_name).compare("add.out") == 0));
 }
+
+// Only add-float kernel.
+// inline constexpr bool should_include_kernel_dtype(
+//     const char* operator_name,
+//     executorch::aten::ScalarType scalar_type) {
+//   return (
+//       (std::string_view(operator_name).compare("add.out") == 0) &&
+//       (scalar_type == executorch::aten::ScalarType::Float));
+// }
+
 #endif
 
 namespace torch {