Run test_cutlass.py error:unexpected keyword argument 'capture_output'

Questions
#1

When I run tests/python/contrib/test_cutlass.py, error:

FAILED …/tests/python/contrib/test_cutlass.py::test_dense - TypeError: init() got an unexpected keyword argument ‘capture_output’

FAILED …/tests/python/contrib/test_cutlass.py::test_dense_bias - TypeError: init() got an unexpected keyword argument ‘capture_output’

FAILED …/tests/python/contrib/test_cutlass.py::test_dense_bias_relu - TypeError: init() got an unexpected keyword argument ‘capture_output’

FAILED …/tests/python/contrib/test_cutlass.py::test_dense_bias_gelu - TypeError: init() got an unexpected keyword argument ‘capture_output’

Thanks if you can help me!

#2

Oh right, I forgot to fix that. This happens on Python 3.6 or older, so can you try 3.7 or 3.8?

#3

Thanks, I ran it successful in python3.7.

I see all the examples about cutlass is about dense in TVM.

Is it currently possible to run convolution via Cutlass in TVM?

#4

Not yet, but I’ll work on conv2d support next. Stay tuned.

#5

Thanks, I’m very interested in this feature and look forward to your work.

#6

Now conv2d is fully supported, including residual block fusion.