From 991bd9fc60c22fc51dab835a15c2615796b36399 Mon Sep 17 00:00:00 2001 From: Ramana Radhakrishnan 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 a7e55320d6dd..7cb40417d2cc 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")