ading
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!
masahi
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?
ading
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?
masahi
4
Not yet, but I’ll work on conv2d support next. Stay tuned.
ading
5
Thanks, I’m very interested in this feature and look forward to your work.
masahi
6
Now conv2d is fully supported, including residual block fusion.