update setup.py

BUPT-GAMMA · May 15, 2024 · 3977d29 · 3977d29
1 parent f1072cf
commit 3977d29
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -101,7 +101,7 @@ def load_ops_extensions():
                 extensions.append(PyCudaExtension(
                     name=osp.join(ops_dir, f'_{ops_prefix}_cuda').replace(osp.sep, "."),
                     sources=[osp.join(src_dir, f) for f in src_files],
-                    include_dirs=[ops.abspath(osp.join('third_party', d)) for d in ops_third_party_deps[i]],
+                    include_dirs=[osp.abspath(osp.join('third_party', d)) for d in ops_third_party_deps[i]],
                     extra_compile_args=['-std=c++17']
                 ))