Add GPU architectures to the build-info file (#1840)

Add semicolon-separated list of target GPU archs as `gpu_architectures` version info property --------- Signed-off-by: Partho Sarthi <[email protected]>
NVIDIA · Mar 7, 2024 · 8f8aeed · 8f8aeed
1 parent 0af4d0d
commit 8f8aeed
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/build/build-info b/build/build-info
@@ -1,7 +1,7 @@
 #!/usr/bin/env bash
 
 #
-# Copyright (c) 2022, NVIDIA CORPORATION. All rights reserved.
+# Copyright (c) 2022-2024, NVIDIA CORPORATION. All rights reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -20,18 +20,21 @@
 # Arguments:
 #   version  - The current version of the project
 #   git_path - The path to the repository
+#   libcudf_path - The path to the libcudf library
 set -e
 
 echo_build_properties() {
   version=$1
   git_path=$2
-  shift 2
+  libcudf_path=$3
+  shift 3
   echo version=$version
   echo user=$USER
   echo revision=$(cd "$git_path" && git rev-parse HEAD)
   echo branch=$(cd "$git_path" && git rev-parse --abbrev-ref HEAD)
   echo date=$(date -u +%Y-%m-%dT%H:%M:%SZ)
   echo url=$(cd "$git_path" && git config --get remote.origin.url)
+  echo gpu_architectures=$(cuobjdump "$libcudf_path" 2>/dev/null | awk -F_ '/arch =/ {print $2}' | sort -n -u | tr '\n' ';')
   for arg in "$@"; do
     echo $arg
   done

diff --git a/pom.xml b/pom.xml
@@ -476,13 +476,15 @@
                   <arg value="${project.basedir}/build/build-info"/>
                   <arg value="${project.version}"/>
                   <arg value="${cudf.path}"/>
+                  <arg value="${libcudf.build.path}/libcudf.a"/>
                 </exec>
                 <exec executable="bash"
                       output="${project.build.directory}/extra-resources/spark-rapids-jni-version-info.properties"
                       failonerror="true">
                   <arg value="${project.basedir}/build/build-info"/>
                   <arg value="${project.version}"/>
                   <arg value="${project.basedir}"/>
+                  <arg value="${native.build.path}/libcudf.so"/>
                 </exec>
               </target>
             </configuration>