We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 40a1597 commit 78e707dCopy full SHA for 78e707d
setup.py
@@ -49,9 +49,8 @@ def cuaev_extension_kwargs(
49
print(f"Will build cuAEV with support for SMs: {', '.join(sms)}")
50
51
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
+ # The following defs are not necessarily required to use CUB safely, but are
+ # harmless
55
nvcc_args.extend(
56
[
57
"-DCUB_NS_QUALIFIER=::cuaev::cub",
0 commit comments