Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

tiny-cuda-nn: fix a couple of build issues #359664

Merged
merged 1 commit into from
Dec 5, 2024

Conversation

al3xtjames
Copy link
Contributor

The CUDA architecture detection code in tiny-cuda-nn's Torch bindings convert CUDA capabilities to ints, which breaks with 9.0a. Filter 9.0a from the CUDA capabilities passed to TCNN_CUDA_ARCHITECTURES. Also patch the Torch bindings to replace the hardcoded -std=c++14 argument to nvcc with -std=c++17 for CUDA 11+ to fix the build.

Things done

  • Built on platform(s)
    • x86_64-linux
    • aarch64-linux
    • x86_64-darwin
    • aarch64-darwin
  • For non-Linux: Is sandboxing enabled in nix.conf? (See Nix manual)
    • sandbox = relaxed
    • sandbox = true
  • Tested, as applicable:
  • Tested compilation of all packages that depend on this change using nix-shell -p nixpkgs-review --run "nixpkgs-review rev HEAD". Note: all changes have to be committed, also see nixpkgs-review usage
  • Tested basic functionality of all binary files (usually in ./result/bin/)
  • 25.05 Release Notes (or backporting 24.11 and 25.05 Release notes)
    • (Package updates) Added a release notes entry if the change is major or breaking
    • (Module updates) Added a release notes entry if the change is significant
    • (Module addition) Added a release notes entry if adding a new NixOS module
  • Fits CONTRIBUTING.md.

Add a 👍 reaction to pull requests you find important.

The CUDA architecture detection code in tiny-cuda-nn's Torch bindings
convert CUDA capabilities to `int`s, which breaks with 9.0a. Filter
9.0a from the CUDA capabilities passed to TCNN_CUDA_ARCHITECTURES. Also
patch the Torch bindings to replace the hardcoded `-std=c++14` argument
to nvcc with `-std=c++17` for CUDA 11+ to fix the build.
@ofborg ofborg bot requested a review from ConnorBaker November 28, 2024 13:59
@ofborg ofborg bot added 10.rebuild-darwin: 0 This PR does not cause any packages to rebuild on Darwin 10.rebuild-linux: 1-10 labels Nov 28, 2024
@ConnorBaker
Copy link
Contributor

Tentatively I’ll say this looks good; I haven’t had a chance to build or test running this yet.
@al3xtjames were you able to build and run samples to verify the program works at runtime?

@ConnorBaker ConnorBaker added the 6.topic: cuda Parallel computing platform and API label Dec 1, 2024
@al3xtjames
Copy link
Contributor Author

I was able to run python3 samples/mlp_learning_an_image_pytorch.py after applying the fix in NVlabs/tiny-cuda-nn#463.

@ConnorBaker ConnorBaker self-assigned this Dec 2, 2024
@wegank wegank added 12.approved-by: package-maintainer This PR was reviewed and approved by a maintainer listed in the package 12.approvals: 1 This PR was reviewed and approved by one reputable person labels Dec 2, 2024
@ConnorBaker ConnorBaker merged commit 833953e into NixOS:master Dec 5, 2024
38 of 39 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
6.topic: cuda Parallel computing platform and API 10.rebuild-darwin: 0 This PR does not cause any packages to rebuild on Darwin 10.rebuild-linux: 1-10 12.approvals: 1 This PR was reviewed and approved by one reputable person 12.approved-by: package-maintainer This PR was reviewed and approved by a maintainer listed in the package
Projects
Status: Done
Development

Successfully merging this pull request may close these issues.

3 participants