Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

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

Open
IzanCatalan opened this issue Dec 10, 2024 · 2 comments
Open

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

IzanCatalan opened this issue Dec 10, 2024 · 2 comments
Labels
? - Needs Triage question Question

Comments

@IzanCatalan
Copy link

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.
@jackkosaian
Copy link
Contributor

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

@IzanCatalan
Copy link
Author

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
? - Needs Triage question Question
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@jackkosaian @IzanCatalan