Only include virtual architecture cuda builds for ci wheels

Q-Minh · Oct 18, 2024 · 0ec18e3 · 0ec18e3
1 parent 7cbe015
commit 0ec18e3
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 8 deletions.
diff --git a/CMakePresets.json b/CMakePresets.json
@@ -190,13 +190,22 @@
                 }
             }
         },
+        {
+            "name": "local-pip-cuda",
+            "inherits": [
+                "pip-cuda"
+            ],
+            "cacheVariables": {
+                "CMAKE_CUDA_ARCHITECTURES": "native"
+            }
+        },
         {
             "name": "ci-pip-cuda",
             "inherits": [
                 "pip-cuda"
             ],
             "cacheVariables": {
-                "CMAKE_CUDA_ARCHITECTURES": "80"
+                "CMAKE_CUDA_ARCHITECTURES": "70-virtual"
             }
         }
     ]

diff --git a/source/CMakeLists.txt b/source/CMakeLists.txt
@@ -95,13 +95,6 @@ if(PBAT_USE_CUDA)
         CUDA_RESOLVE_DEVICE_SYMBOLS ON
     )
 
-    if(NOT DEFINED CMAKE_CUDA_ARCHITECTURES)
-        set_target_properties(PhysicsBasedAnimationToolkit_PhysicsBasedAnimationToolkit
-            PROPERTIES
-            CUDA_ARCHITECTURES "native"
-        )
-    endif()
-
     get_target_property(_pbat_sources PhysicsBasedAnimationToolkit_PhysicsBasedAnimationToolkit SOURCES)
 
     foreach(_source IN ITEMS ${_pbat_sources})