build for 2.2.1 in CI

chengzeyi · Mar 7, 2024 · 9237048 · 9237048
1 parent e0728bc
commit 9237048
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/.github/workflows/wheels.yml b/.github/workflows/wheels.yml
@@ -32,6 +32,7 @@ jobs:
           - "3.11"
         torch_version:
           - "2.2.0"
+          - "2.2.1"
         cuda_short_version:
           - "118"
           - "121"

diff --git a/src/sfast/csrc/operators/cudnn/cudnn_convolution_impl.cc b/src/sfast/csrc/operators/cudnn/cudnn_convolution_impl.cc
@@ -1185,12 +1185,12 @@ select_conv_backend(const Tensor &input_r, const Tensor &weight_r,
                     int64_t groups_) {
   return torch::native::select_conv_backend(
       input_r, weight_r, bias_opt,
-#if TORCH_VERSION_MINOR >= 2
+#if TORCH_VERSION_MAJOR == 2 && TORCH_VERSION_MINOR >= 2 || TORCH_VERSION_MAJOR > 2
       fromIntArrayRefUnchecked
 #endif
       (stride_),
       fromIntArrayRefUnchecked(padding_),
-#if TORCH_VERSION_MINOR >= 2
+#if TORCH_VERSION_MAJOR == 2 && TORCH_VERSION_MINOR >= 2 || TORCH_VERSION_MAJOR > 2
       fromIntArrayRefUnchecked
 #endif
       (dilation_),
-Original file line number
+Diff line change
@@ Expand Up / @@ -32,6 +32,7 @@ jobs: @@
               - "3.11"
             torch_version:
               - "2.2.0"
+              - "2.2.1"
             cuda_short_version:
               - "118"
               - "121"
@@ Expand Down @@