diff --git a/setup.py b/setup.py index fec5036..b39efd0 100644 --- a/setup.py +++ b/setup.py @@ -73,7 +73,7 @@ if __name__ == '__main__': packages=['deep_gemm', 'deep_gemm/jit', 'deep_gemm/jit_kernels'], package_data={ 'deep_gemm': [ - 'include/deep_gemm/**/*', + 'include/deep_gemm/*', 'include/cute/**/*', 'include/cutlass/**/*', ]