Skip to content

Commit 74d813f

Browse files
aakhundovfacebook-github-bot
authored andcommitted
Remove internal cub dependency (facebookincubator#943)
Summary: Pull Request resolved: facebookincubator#943 Following the investigation in D49853192, it has become clear that the internal cub dependency in the `AITemplate/fb/3rdparty` is not actually used during AIT-driven compilation. Consequently, in this diff we're trying to remove the internal cub dependency. Reviewed By: khabinov, chenyang78 Differential Revision: D50011260 fbshipit-source-id: 41c592734e7facd1b61f852146d32c7cca415571
1 parent 0a67de1 commit 74d813f

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

python/aitemplate/backend/cuda/target_def.py

-3
Original file line numberDiff line numberDiff line change
@@ -286,7 +286,6 @@ def __init__(self, arch="80", remote_cache_bytes=None, **kwargs):
286286
cutlass_src_path = parutil.get_dir_path(
287287
"aitemplate/AITemplate/fb/3rdparty/cutlass"
288288
)
289-
cub_src_path = parutil.get_dir_path("aitemplate/AITemplate/fb/3rdparty/cub")
290289
static_files_path = parutil.get_dir_path("aitemplate/AITemplate/static")
291290
if "optimize_for_compilation_time" in kwargs:
292291
FBCUDA.optimize_for_compilation_time_ = kwargs[
@@ -302,9 +301,7 @@ def __init__(self, arch="80", remote_cache_bytes=None, **kwargs):
302301
self._include_path = tempfile.mkdtemp()
303302

304303
FBCUDA.cutlass_path_ = self._include_path + "/cutlass"
305-
self.cub_path_ = self._include_path + "/cub"
306304
shutil.copytree(cutlass_src_path, FBCUDA.cutlass_path_)
307-
shutil.copytree(cub_src_path, self.cub_path_)
308305

309306
attention_src_path = parutil.get_dir_path(
310307
"aitemplate/AITemplate/python/aitemplate/backend/cuda/attention/src"

0 commit comments

Comments
 (0)