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

ading · November 4, 2021, 5:04am

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!

masahi · November 4, 2021, 5:39am

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

ading · November 4, 2021, 6:55am

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?

masahi · November 4, 2021, 7:41am

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

ading · November 4, 2021, 8:17am

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

masahi · January 6, 2022, 2:43am

Now conv2d is fully supported, including residual block fusion.