Add Blackwell support (#265)

* Add Blackwell support

* Update CHANGELOG

Author: Alexey-Kamenev

CHANGELOG.md

@@ -10,6 +10,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Added
 
+- Added Blackwell support
+
 ### Changed
 
 ### Deprecated
@@ -30,7 +32,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Improved performance of connectivity calculation. Refactor of Least squares
   gradients
 - Added Incompressible Navier Stokes PDE as a special formulation
-- Added ability for `make_nodes` function to return a dict 
+- Added ability for `make_nodes` function to return a dict
 
 ### Changed
 

setup.py

@@ -33,6 +33,8 @@ def cuda_extension():
         nvcc_args.append("-gencode=arch=compute_86,code=sm_86")
     if cuda_version >= 12:
         nvcc_args.append("-gencode=arch=compute_90,code=sm_90")
+    if cuda_version >= 12.8:
+        nvcc_args.append("-gencode=arch=compute_100,code=sm_100")
 
     nvcc_args.append("-t=0")  # Enable multi-threaded builds
     # nvcc_args.append("--time=output.txt")