From c44a47422009c42e9d07de4ab1fe5f8c278eb701 Mon Sep 17 00:00:00 2001 From: Zihao Ye Date: Sun, 16 Jul 2023 00:33:30 -0700 Subject: [PATCH] [Package] Remove cutlass media/docs inside cutlass_fpA_intB_gemm (#15328) --- python/setup.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/python/setup.py b/python/setup.py index b4c4c40943b08..0f1d0b44363a3 100644 --- a/python/setup.py +++ b/python/setup.py @@ -95,6 +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" + ) + ) break else: libs = None