Commit 1f989ba0 authored by yan.yan's avatar yan.yan
Browse files

fix windows cuda 11.3 problem

parent ec9c88d3
......@@ -16,7 +16,7 @@ jobs:
strategy:
matrix:
python-version: ['3.7', '3.8', '3.9', '3.10']
cuda-version: ['10.2', '11.1', '11.4']
cuda-version: ['10.2', '11.1', '11.3', '11.4']
steps:
- uses: actions/checkout@master
- uses: dorny/paths-filter@v2
......
[build-system]
requires = ["setuptools>=41.0", "wheel", "pccm>=0.2.19", "cumm>=0.2.0"]
requires = ["setuptools>=41.0", "wheel", "pccm>=0.2.21", "cumm>=0.2.1"]
build-backend = "setuptools.build_meta"
......@@ -29,7 +29,8 @@ class CustomThrustLib(pccm.Class):
super().__init__()
self.add_dependency(ThrustLib)
# https://github.com/NVIDIA/thrust/issues/1401#issuecomment-806403746
self.build_meta.add_cflags("nvcc", "-Xcompiler", "-fno-gnu-unique")
if compat.InLinux:
self.build_meta.add_cflags("nvcc", "-Xcompiler", "-fno-gnu-unique")
class ThrustCustomAllocatorV2(pccm.Class, pccm.pybind.PybindClassMixin):
def __init__(self):
......
......@@ -28,7 +28,7 @@ if (($CUDA_VERSION_FULL -eq "10.2") -or ($CUDA_VERSION_FULL -eq "11.0") -or ($CU
)
} elseif ($CUDA_VERSION_FULL -eq "11.3"){
$CUDA_PACKAGES_IN = @(
"nvcc";
"cuda_nvcc";
"visual_studio_integration";
"cuda_nvrtc";
"cuda_cudart";
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment