build-cuda-ext.sh

#!/bin/bash
#
# @File:   build-cuda-ext.sh
# @Author: Haozhe Xie
# @Date:   2024-09-22 10:43:34
# @Last Modified by: Haozhe Xie
# @Last Modified at: 2024-09-22 10:46:29
# @Email:  [email protected]

SCRIPT_DIR=$(dirname "$(readlink -f "$0")")
EXTENSIONS_DIR="$SCRIPT_DIR/citydreamer/extensions"
WHEELS_DIR="$SCRIPT_DIR/wheels"

mkdir -p "$WHEELS_DIR"

for dir in "$EXTENSIONS_DIR"/*/; do
  if [ -d "$dir" ]; then
    echo "Processing $dir"
    (cd "$dir" && pip wheel .)
    mv "$dir"*.whl "$WHEELS_DIR"
  fi
done