From 63cedb640d78392363c1879d03c7b022a50d7541 Mon Sep 17 00:00:00 2001 From: Zihao Date: Sat, 15 Jul 2023 16:08:28 -0700 Subject: [PATCH 1/2] remove cutlass media/docs inside cutlass_fpA_intB_gemm --- python/setup.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/setup.py b/python/setup.py index b4c4c40943b0..1d2172f6c05a 100644 --- a/python/setup.py +++ b/python/setup.py @@ -95,6 +95,8 @@ def get_lib_path(): # remove large files _remove_path(os.path.join(candidate_path, "cutlass", "docs")) _remove_path(os.path.join(candidate_path, "cutlass", "media")) + _remove_path(os.path.join(candidate_path, "cutlass_fpA_intB_gemm", "cutlass", "docs")) + _remove_path(os.path.join(candidate_path, "cutlass_fpA_intB_gemm", "cutlass", "media")) break else: libs = None From 2fa39bf0322fe9359f38c4847acd1442e6a25c8c Mon Sep 17 00:00:00 2001 From: Zihao Date: Sat, 15 Jul 2023 19:13:21 -0700 Subject: [PATCH 2/2] pylint --- python/setup.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/python/setup.py b/python/setup.py index 1d2172f6c05a..0f1d0b44363a 100644 --- a/python/setup.py +++ b/python/setup.py @@ -95,8 +95,14 @@ def get_lib_path(): # remove large files _remove_path(os.path.join(candidate_path, "cutlass", "docs")) _remove_path(os.path.join(candidate_path, "cutlass", "media")) - _remove_path(os.path.join(candidate_path, "cutlass_fpA_intB_gemm", "cutlass", "docs")) - _remove_path(os.path.join(candidate_path, "cutlass_fpA_intB_gemm", "cutlass", "media")) + _remove_path( + os.path.join(candidate_path, "cutlass_fpA_intB_gemm", "cutlass", "docs") + ) + _remove_path( + os.path.join( + candidate_path, "cutlass_fpA_intB_gemm", "cutlass", "media" + ) + ) break else: libs = None