@@ -163,7 +163,7 @@ test_cmake_select_ops_in_yaml() {
163163
164164test_cmake_select_ops_in_dict () {
165165 echo " Exporting MobilenetV2"
166- ${PYTHON_EXECUTABLE} -m examples.portable.scripts.export --model_name=" add "
166+ ${PYTHON_EXECUTABLE} -m examples.portable.scripts.export --model_name=" mv2 "
167167
168168 local example_dir=examples/selective_build
169169 local build_dir=cmake-out/${example_dir}
@@ -184,12 +184,12 @@ test_cmake_select_ops_in_dict() {
184184 echo ' Running selective build test'
185185 ${build_dir} /selective_build_test --model_path=" ./mv2.pte"
186186
187- echo " Removing mv2.pte"
188- rm " ./mv2.pte"
187+ # echo "Removing mv2.pte"
188+ # rm "./mv2.pte"
189189}
190190
191191test_cmake_select_ops_in_model () {
192- local model_name=" mv3 "
192+ local model_name=" mv2 "
193193 local model_export_name=" ${model_name} .pte"
194194 echo " Exporting ${model_name} "
195195 ${PYTHON_EXECUTABLE} -m examples.portable.scripts.export --model_name=" ${model_name} "
@@ -210,7 +210,7 @@ test_cmake_select_ops_in_model() {
210210
211211 strip ${build_dir} /selective_build_test
212212 echo $( stat --format=%s ${build_dir} /selective_build_test)
213- rm " ./${model_export_name} "
213+ # rm "./${model_export_name}"
214214}
215215
216216if [[ -z $BUCK ]];
234234 # test_cmake_select_all_ops
235235 # test_cmake_select_ops_in_list
236236 # test_cmake_select_ops_in_yaml
237- test_cmake_select_ops_in_model
238- # test_cmake_select_ops_in_dict
237+ test_cmake_select_ops_in_dict
238+ # test_cmake_select_ops_in_model
239239elif [[ $1 == " buck2" ]];
240240then
241241 test_buck2_select_all_ops
0 commit comments