From ad2ee9735120ec6879e90918b6cd099ec16b9f6d Mon Sep 17 00:00:00 2001
From: Ramana Radhakrishnan <ramana.radhakrishnan@arm.com>
Date: Mon, 11 May 2020 20:00:26 +0100
Subject: [PATCH] Fix topi test for tensorcore (#5563)

---
 topi/tests/python/test_topi_conv2d_nhwc_winograd.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/topi/tests/python/test_topi_conv2d_nhwc_winograd.py b/topi/tests/python/test_topi_conv2d_nhwc_winograd.py
index a7e5532..7cb4041 100644
--- a/topi/tests/python/test_topi_conv2d_nhwc_winograd.py
+++ b/topi/tests/python/test_topi_conv2d_nhwc_winograd.py
@@ -137,6 +137,9 @@ def test_conv2d_nhwc_winograd_direct():
 
 def test_conv2d_nhwc_winograd_tensorcore():
     """Test the conv2d with winograd for nhwc layout"""
+    if not tvm.gpu(0).exist or not tvm.runtime.enabled("cuda"):
+        print("skip because cuda is not enabled..")
+        return
     if not nvcc.have_tensorcore(tvm.gpu(0).compute_version):
         return
     verify_conv2d_nhwc(8,  64, 56,  64, 3, 1, 1, bgemm="tensorcore")
-- 
2.7.4