don't rely on cudnn for compilation (#10495)

apache · Mar 4, 2022 · db8cf2f · db8cf2f
1 parent 3f96f3d
commit db8cf2f
Showing 1 changed file with 16 additions and 12 deletions.
diff --git a/python/tvm/contrib/cudnn.py b/python/tvm/contrib/cudnn.py
@@ -741,18 +741,22 @@ def conv_backward_data(
         tensor_format, pad, stride, dilation, dy.shape, w.shape, output_padding, groups
     )
 
-    algo = conv_backward_data_find_algo(
-        tensor_format,
-        pad,
-        stride,
-        dilation,
-        list(dy.shape),
-        list(w.shape),
-        dx_shape,
-        dy.dtype,
-        conv_dtype,
-        groups,
-    )
+    if exists():
+        # When cudnn exists, find the backward data algo
+        algo = conv_backward_data_find_algo(
+            tensor_format,
+            pad,
+            stride,
+            dilation,
+            list(dy.shape),
+            list(w.shape),
+            dx_shape,
+            dy.dtype,
+            conv_dtype,
+            groups,
+        )
+    else:
+        algo = 1
 
     return te.extern(
         dx_shape,