cutlass icon indicating copy to clipboard operation
cutlass copied to clipboard

[QST] Integer Data Types are available for Conv2d fprop?

Open IzanCatalan opened this issue 1 year ago • 3 comments

What is your question? Hi, I would like to know if I can perform a conv2d fprop with int8 data type, in c++ kernels and in python.

Thanks.

IzanCatalan avatar Dec 10 '24 16:12 IzanCatalan

Yes, it is possible. For example, if you configure CUTLASS with cmake .. -DCUTLASS_NVCC_ARCHS="80", you can see an INT8 Conv2d kernel emitted under tools/library/generated/conv2d/80/i16832fprop_optimized_s8/cutlass_tensorop_i16832fprop_optimized_s8_256x128_64x3_nhwc_align16.cu

jackkosaian avatar Dec 11 '24 20:12 jackkosaian

@jackkosaian are they available in Python? I couldn't check that as I have written in a new post

IzanCatalan avatar Dec 17 '24 11:12 IzanCatalan

This issue has been labeled inactive-30d due to no recent activity in the past 30 days. Please close this issue if no further response or action is needed. Otherwise, please respond with a comment indicating any updates or changes to the original issue and/or confirm this issue still needs to be addressed. This issue will be labeled inactive-90d if there is no activity in the next 60 days.

github-actions[bot] avatar Jan 16 '25 12:01 github-actions[bot]

This issue has been labeled inactive-90d due to no recent activity in the past 90 days. Please close this issue if no further response or action is needed. Otherwise, please respond with a comment indicating any updates or changes to the original issue and/or confirm this issue still needs to be addressed.

github-actions[bot] avatar Apr 16 '25 12:04 github-actions[bot]