Skip to content

Commit 78e707d

Browse files
Fix comment
1 parent 40a1597 commit 78e707d

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

setup.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -49,9 +49,8 @@ def cuaev_extension_kwargs(
4949
print(f"Will build cuAEV with support for SMs: {', '.join(sms)}")
5050

5151
nvcc_args = ["--expt-extended-lambda"]
52-
# The following defs are required to use CUB safely. For details see:
53-
# https://github.com/pytorch/pytorch/pull/55292
54-
# https://github.com/pytorch/pytorch/pull/66219
52+
# The following defs are not necessarily required to use CUB safely, but are
53+
# harmless
5554
nvcc_args.extend(
5655
[
5756
"-DCUB_NS_QUALIFIER=::cuaev::cub",

0 commit comments

Comments
 (0)