Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
torch-spline-conv
Commits
e44831b0
Commit
e44831b0
authored
Oct 22, 2021
by
rusty1s
Browse files
[ci skip]
parent
0483feab
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
2 deletions
+2
-2
.github/workflows/cuda/cu113-Windows.sh
.github/workflows/cuda/cu113-Windows.sh
+1
-1
setup.py
setup.py
+1
-1
No files found.
.github/workflows/cuda/cu113-Windows.sh
View file @
e44831b0
...
...
@@ -13,6 +13,6 @@ export CUDA_FILE=cuda_${CUDA_SHORT}.0_465.89_win10.exe
curl
-k
-L
"
${
CUDA_URL
}
/
${
CUDA_FILE
}
"
--output
"
${
CUDA_FILE
}
"
echo
""
echo
"Installing from
${
CUDA_FILE
}
..."
PowerShell
-Command
"Start-Process -FilePath
\"
${
CUDA_FILE
}
\"
-ArgumentList
\"
-s nvcc_
${
CUDA_SHORT
}
cuobjdump_
${
CUDA_SHORT
}
nvprune_
${
CUDA_SHORT
}
cupti_
${
CUDA_SHORT
}
cublas_dev_
${
CUDA_SHORT
}
cudart_
${
CUDA_SHORT
}
cufft_dev_
${
CUDA_SHORT
}
curand_dev_
${
CUDA_SHORT
}
cusolver_dev_
${
CUDA_SHORT
}
cusparse_dev_
${
CUDA_SHORT
}
npp_dev_
${
CUDA_SHORT
}
nvrtc_dev_
${
CUDA_SHORT
}
nvml_dev_
${
CUDA_SHORT
}
\"
-Wait -NoNewWindow"
PowerShell
-Command
"Start-Process -FilePath
\"
${
CUDA_FILE
}
\"
-ArgumentList
\"
-s nvcc_
${
CUDA_SHORT
}
cuobjdump_
${
CUDA_SHORT
}
nvprune_
${
CUDA_SHORT
}
cupti_
${
CUDA_SHORT
}
cublas_dev_
${
CUDA_SHORT
}
cudart_
${
CUDA_SHORT
}
cufft_dev_
${
CUDA_SHORT
}
curand_dev_
${
CUDA_SHORT
}
cusolver_dev_
${
CUDA_SHORT
}
cusparse_dev_
${
CUDA_SHORT
}
thrust_
${
CUDA_SHORT
}
npp_dev_
${
CUDA_SHORT
}
nvrtc_dev_
${
CUDA_SHORT
}
nvml_dev_
${
CUDA_SHORT
}
\"
-Wait -NoNewWindow"
echo
"Done!"
rm
-f
"
${
CUDA_FILE
}
"
setup.py
View file @
e44831b0
...
...
@@ -29,7 +29,7 @@ def get_extensions():
for
main
,
suffix
in
product
(
main_files
,
suffices
):
define_macros
=
[]
extra_compile_args
=
{
'cxx'
:
[
'-O2'
]}
extra_link_args
=
[]
extra_link_args
=
[
'-s'
]
if
suffix
==
'cuda'
:
define_macros
+=
[(
'WITH_CUDA'
,
None
)]
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment