fix #15 completely remove cuflit

FindDefinition · Oct 19, 2022 · c91f779 · c91f779
1 parent d202989
commit c91f779
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/cumm/tensorview_bind.py b/cumm/tensorview_bind.py
@@ -51,8 +51,10 @@ def __init__(self):
             cuda_ver = get_cuda_version_by_nvcc().split(".")
             cuda_ver_ints = list(map(int, cuda_ver))
             if cuda_ver_ints >= [11, 4]:
-                self.add_include("nv_decode.h")
-                self.build_meta.add_libraries("nvrtc", "cufilt")
+                # cuflit has problem in arm platforms and windows, so we remove it.
+                # self.add_include("nv_decode.h")
+                # self.build_meta.add_libraries("nvrtc", "cufilt")
+                self.build_meta.add_libraries("nvrtc")
             else:
                 self.build_meta.add_libraries("nvrtc")
             if compat.InLinux: