Add mmcv specific patch

MiroPsota · Dec 4, 2024 · 3036ef3 · 3036ef3
1 parent eab4377
commit 3036ef3
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git 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
@@ -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'):