Fix the bug

microsoft · Jun 21, 2024 · cdf0d62 · cdf0d62
1 parent 0ed30db
commit cdf0d62
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxruntime/core/providers/cuda/tensor/upsample.cc b/onnxruntime/core/providers/cuda/tensor/upsample.cc
@@ -230,7 +230,7 @@ Status Upsample<T>::BaseCompute(OpKernelContext* context,
           }
 
           const bool is_2D = X_dims.size() == 2;
-          const bool is_nchw = is_2D ? true : scales[1] == 1.0f;
+          const bool is_nchw = is_2D ? true : (scales[0] == 1.0f && scales[1] == 1.0f);
 
           ORT_RETURN_IF_NOT(is_nchw,
                             "Resize 'Cubic' mode only supports NCWH layout "