PyTorch fp8 precision gemm lib w/ cutlass, this is inspired by aredden/torch-cublas-hgemm transfering its HGEMM to cutlass and get FP8 GEMM working with cutlass
figure out how to get nf4 linear layers to cutlass
-
Notifications
You must be signed in to change notification settings - Fork 0
PyTorch fp8 precision gemm lib w/ cutlass
License
shauray8/cutlass_q8gemm
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
PyTorch fp8 precision gemm lib w/ cutlass
Resources
License
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published