wip

pytorch · Nov 23, 2024 · dc0b413 · dc0b413
1 parent b31fe6b
commit dc0b413
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cmake/modules/GpuCppLibrary.cmake b/cmake/modules/GpuCppLibrary.cmake
@@ -146,7 +146,7 @@ function(gpu_cpp_library)
 
     set(flags)
     set(singleValueArgs
-        PREFIX          # Desired name for the library target (and simultaneously, the prefix for intermediate targets)
+        PREFIX          # Desired name for the library target (and by extension, the prefix for naming intermediate targets)
         TYPE            # Target type, e.g., MODULE, OBJECT.  See https://cmake.org/cmake/help/latest/command/add_library.html
     )
     set(multiValueArgs
@@ -196,7 +196,7 @@ function(gpu_cpp_library)
     # Build the Library
     ############################################################################
 
-    set(lib_name ${args_PREFIX}_py)
+    set(lib_name ${args_PREFIX})
     if(USE_ROCM)
         # Fetch the equivalent HIPified sources if available.
         # This presumes that `hipify()` has already been run.