File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -245,7 +245,8 @@ def get_sources(mod_name):
245245 extra_link_args = []
246246 extra_cythonize_kwargs = {}
247247 if sys .platform == "win32" :
248- extra_compile_args += ["/std:c++17" ]
248+ # CCCL headers #error without the conforming MSVC preprocessor.
249+ extra_compile_args += ["/std:c++17" , "/Zc:preprocessor" ]
249250 if debug :
250251 raise RuntimeError ("Debuggable builds are not supported on Windows." )
251252 else :
Original file line number Diff line number Diff line change @@ -686,6 +686,8 @@ cdef class TensorMapDescriptor:
686686 # TODO(seberg): CCCL has various limitations as of writing, see:
687687 # https://github.com/NVIDIA/cccl/issues/10511.
688688 # Because of this, the path is only taken in _very_ narrow conditions.
689+ # We should probably use it broadly as soon as the fix is available in
690+ # CCCL 3.5.0 (and likely just not bother with these narrow conditions).
689691 make_tma = _get_cccl_make_tma_fn()
690692 if (make_tma != NULL
691693 and device_type == _kDLCUDA
Original file line number Diff line number Diff line change 1- # This file was generated by stubgen-pyx v0.2.6 from cuda_core/cuda/core/_tensor_map_cccl.pyx
1+ # This file was generated by stubgen-pyx v0.2.19 from cuda_core/cuda/core/_tensor_map_cccl.pyx
22
33"""Optional CCCL <cuda/tma> helper for TensorMapDescriptor.
44
@@ -9,4 +9,3 @@ shared cudart. It must not be imported at ``cuda.core`` package import time.
99``get_make_tma_descriptor_tiled`` is exported via ``__pyx_capi__`` for
1010soft-linking from ``_tensor_map.pyx`` (returns real fn or NULL).
1111"""
12- from __future__ import annotations
You can’t perform that action at this time.
0 commit comments