[Package] Remove cutlass media/docs inside cutlass_fpA_intB_gemm (apa…

…che#15328)
junrushao · Jul 27, 2023 · 5eb420a · 5eb420a
1 parent 9332888
commit 5eb420a
Showing 1 changed file with 8 additions and 0 deletions.
diff --git 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