Unverified Commit e9dd9da6 authored by machenmusik's avatar machenmusik Committed by GitHub
Browse files

avoid Windows repeated CUDA setup (#150)



restore working 0.2.1 version, since .so will not exist on Windows
Co-authored-by: default avatarmchen001_comcast <Michael_Chen@cable.comcast.com>
parent 58d6b4c4
......@@ -64,9 +64,12 @@ if cuda_toolkit_available():
# f"download cub because the cuda version is {cuda_toolkit_version()}"
# )
if os.path.exists(os.path.join(build_dir, f"{name}.so")):
if os.listdir(build_dir) != []:
# If the build exists, we assume the extension has been built
# and we can load it.
Console().print(
"[yellow]NerfAcc: CUDA set up, loading (should be quick)[/yellow]"
)
_C = load(
name=name,
sources=glob.glob(os.path.join(PATH, "csrc/*.cu")),
......
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