From 3036ef31a43e88d4f5c8a039efafd132f2376fa0 Mon Sep 17 00:00:00 2001 From: Miroslav Psota <9453736+MiroPsota@users.noreply.github.com> Date: Tue, 3 Dec 2024 22:22:29 +0100 Subject: [PATCH] Add mmcv specific patch --- package_specific.sh | 6 ++++++ package_specific/mmcv_cpp14.patch | 8 ++++++++ 2 files changed, 14 insertions(+) create mode 100644 package_specific/mmcv_cpp14.patch diff --git a/package_specific.sh b/package_specific.sh index d6465d6..a805929 100644 --- a/package_specific.sh +++ b/package_specific.sh @@ -48,3 +48,9 @@ if [[ $REPO == "facebookresearch/fairseq" ]]; then pip install cython patch -p0 < "$SCRIPT_DIR"/package_specific/fairseq_cub.patch fi + +if [[ $REPO == "open-mmlab/mmcv" ]] \ + && [[ $TORCH_VERSION == "1.12.1" ]] \ + && [[ $COMPUTE_PLATFORM == "cu102" ]]; then + patch -p0 < "$SCRIPT_DIR"/package_specific/mmcv_cpp14.patch +fi diff --git a/package_specific/mmcv_cpp14.patch b/package_specific/mmcv_cpp14.patch new file mode 100644 index 0000000..ebeecf0 --- /dev/null +++ b/package_specific/mmcv_cpp14.patch @@ -0,0 +1,8 @@ +--- setup.py ++++ setup.py +@@ -204 +204 @@ +- if parse_version(torch.__version__) <= parse_version('1.12.1'): ++ if parse_version(torch.__version__) <= parse_version('1.12.2'): +@@ -441 +441 @@ +- if parse_version(torch.__version__) <= parse_version('1.12.1'): ++ if parse_version(torch.__version__) <= parse_version('1.12.2'):