diff --git a/etc/bash-utils.sh b/etc/bash-utils.sh index 6841a57..9c090af 100755 --- a/etc/bash-utils.sh +++ b/etc/bash-utils.sh @@ -651,7 +651,7 @@ clean-rmm-python() { find "$RMM_HOME" -type f -name '*.pyc' -delete; find "$RMM_HOME" -type d -name '__pycache__' -delete; find "$RMM_HOME/python" -type f -name '*.so' -delete; - find "$RMM_HOME/python" -type f -name '*.cpp' -delete; + find "$RMM_HOME/python" -type f -name '*.cpp' -not -name "_torch_allocator.cpp" -delete; } export -f clean-rmm-python;