Skip to content

Commit

Permalink
Don't delete RMM's _torch_allocator.cpp
Browse files Browse the repository at this point in the history
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.
  • Loading branch information
gmarkall committed Dec 20, 2023
1 parent 11107b1 commit cffdde2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion etc/bash-utils.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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;
Expand Down

0 comments on commit cffdde2

Please sign in to comment.