leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 26 Packages Projects Releases Wiki Activity

add the missing pieces (#392)

Browse Source 
Co-authored-by: Haicheng Wu <haichengw@nvidia.com>
This commit is contained in:
Haicheng Wu
2021-12-25 04:29:54 -08:00
committed by GitHub
co-authored by Haicheng Wu
parent dceabd4c5a
commit f78994bb40
3 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/library/scripts/generator.py
+3
View File
@@ -2641,6 +2641,9 @@ def GenerateSM80(manifest, args):
GenerateSM80_TensorOp_1688_fast_math(manifest, args)
GenerateSM80_SparseTensorOp_16816_fast_math(manifest, args)
GenerateSM80_TensorOp_1688_complex(manifest, args)
# 3xTF32
GenerateSM80_TensorOp_1688_fast_fp32_math(manifest, args)
GenerateSM80_TensorOp_1688_fast_fp32_math_complex(manifest, args)
GenerateSM80_TensorOp_884(manifest, args)
GenerateSM80_TensorOp_884_complex(manifest, args)
GenerateSM80_TensorOp_884_complex_gaussian(manifest, args)