added miopenGetConvolutionSpatialDim if ROCm5.5 (#14772)

The API should be `miopenGetConvolutionSpatialDim(cdesc, &spatial_dim)`, NOT `miopenGetConvolutionDescriptorSize(cdesc, &spatial_dim)`
microsoft · Mar 3, 2023 · 6e2ca15 · 6e2ca15
1 parent 0ebe8e3
commit 6e2ca15
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxruntime/contrib_ops/rocm/fused_conv.cc b/onnxruntime/contrib_ops/rocm/fused_conv.cc
@@ -75,7 +75,7 @@ struct FNVHash {
   void HashConvolutionDescriptor(miopenConvolutionDescriptor_t cdesc) {
     int spatial_dim = 1;
 #if ROCM_VERSION >= 50500
-    miopenGetConvolutionDescriptorSize(cdesc, &spatial_dim);
+    miopenGetConvolutionSpatialDim(cdesc, &spatial_dim);
 #else
     // Previous versions of MIOpen doesn't provide API to probe the dimension of a
     // miopenConvolutionDescriptor_t, so we have to guess.