CUTLASS 2.10 updates (#622)

Co-authored-by: Aniket Shivam <ashivam@nvidia.com>
This commit is contained in:
ANIKET SHIVAM
2022-09-12 18:26:30 -07:00
committed by GitHub
parent beae168f90
commit e773429f7e
96 changed files with 8365 additions and 1667 deletions

View File

@ -43,7 +43,7 @@ try:
Pybind11Extension("cutlass",
["src/cpp/cutlass.cpp"],
include_dirs=include_dirs,
extra_compile_args=["-fpermissive"])
extra_compile_args=["-fpermissive", "-w"])
]
except ImportError:
pass
@ -69,7 +69,8 @@ setup(
'typeguard',
'bfloat16',
'typing',
'scikit-build'
'scikit-build',
'treelib'
],
cmdclass={
'rmm': BuildRMM