From cffdde25a0b5e361f29e6becd5ec72e9ef386a3b Mon Sep 17 00:00:00 2001 From: Graham Markall Date: Wed, 20 Dec 2023 15:43:09 +0000 Subject: [PATCH] Don't delete RMM's `_torch_allocator.cpp` The RMM Python source now contains a non-generated C++ file, `_torch_allocator.cpp`, from rapidsai/rmm#1407, so we need to avoid deleting it when cleaning rmm-python. This approach might not be the best if RMM Python goes on to include more C++ sources, but this fixes the clean / build process for now. --- etc/bash-utils.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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;