allow target none

apache · Feb 1, 2022 · a031085 · a031085
1 parent 83e7af7
commit a031085
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/tvm/topi/nn/conv2d_transpose.py b/python/tvm/topi/nn/conv2d_transpose.py
@@ -301,8 +301,8 @@ def conv2d_transpose_legalize(attrs, inputs, types):
     data, kernel = inputs
     kernel_layout = attrs["kernel_layout"]
 
-    target = tvm.target.Target.current(allow_none=False)
-    if "cudnn" in target.libs:
+    target = tvm.target.Target.current(allow_none=True)
+    if target and "cudnn" in target.libs:
         # cuDNN backend can directly operate on NHWC layout.
         return None