"docs/vscode:/vscode.git/clone" did not exist on "17f1432ab2c74bed54df863be48e23b4113cbb37"
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(): ...@@ -64,9 +64,12 @@ if cuda_toolkit_available():
# f"download cub because the cuda version is {cuda_toolkit_version()}" # 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 # If the build exists, we assume the extension has been built
# and we can load it. # and we can load it.
Console().print(
"[yellow]NerfAcc: CUDA set up, loading (should be quick)[/yellow]"
)
_C = load( _C = load(
name=name, name=name,
sources=glob.glob(os.path.join(PATH, "csrc/*.cu")), 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