Merge pull request #1377 from mhalk/mhalk/feat/run-ck-install

[CK] Add CL flag to install Composable Kernels

Author: mhalk

bin/run_composable-kernels.sh

@@ -18,12 +18,15 @@ export PATH=$AOMP/bin:$PATH
 function printHelp {
   echo "Usage: run_composable-kernels.sh"
   echo "  -h: Show this help message"
+  echo "  -i: Install the (incremental) CK build"
   echo "  -r: Rebuild the CK repo"
   echo "  -u: Update the CK repo"
   echo "  -b: Update the CK benchmarks repo"
   exit 0
 }
 
+# Some tests may require an installed instance of CK.
+ShouldInstallCK='no'
 # For some situations during testing it may not be desired to rebuild the CK repo.
 ShouldRebuildCK='no'
 # While doing perf / other compiler work, keeping CK fix is useful.
@@ -32,11 +35,15 @@ ShouldUpdateCKRepo='no'
 # CK Benchmarks is priate, maybe do not want to update it.
 ShouldUpdateCKBenchmarks='no'
 
-while getopts "hrub" opt; do
+while getopts "hirub" opt; do
   case $opt in
   h)
     printHelp
     ;;
+  i)
+    # Install the CK build
+    ShouldInstallCK='yes'
+    ;;
   r)
     # Rebuild the CK repo
     ShouldRebuildCK='yes'
@@ -63,6 +70,7 @@ done
 : ${CK_BENCHMARK_REPO:=$CK_TOP/ck-benchmark}
 # Move this to its own place, to avoid potential permission conflicts with certain setups.
 : ${CK_BENCHMARK_RESULT:=$CK_TOP/ck-benchmark-result}
+: ${CK_INSTALL:=$CK_TOP/ck-install}
 
 # Get some info on the system
 : ${ROCM_PATH:=/opt/rocm}
@@ -89,7 +97,7 @@ elif [ "${ShouldUpdateCKRepo}" == 'yes' ]; then
 fi
 
 # TODO Fix / Finalize the cmake command
-CKCmakeCmd="cmake -GNinja -B ${CK_BUILD} -S ${CK_REPO} -DCMAKE_PREFIX_PATH=${ROCM_PATH} "
+CKCmakeCmd="cmake -GNinja -B ${CK_BUILD} -S ${CK_REPO} -DCMAKE_PREFIX_PATH=${ROCM_PATH} -DCMAKE_INSTALL_PREFIX=${CK_INSTALL} "
 CKCmakeCmd+="-DCMAKE_CXX_COMPILER=${AOMP}/bin/clang++ -DCMAKE_HIP_COMPILER=${AOMP}/bin/clang++ "
 CKCmakeCmd+="-DCMAKE_CXX_COMPILER_LAUNCHER=ccache "
 CKCmakeCmd+="-DCMAKE_BUILD_TYPE=Release -DGPU_TARGETS=${CK_GPU_TARGETS}"
@@ -116,6 +124,23 @@ if [ "${ShouldRebuildCK}" == 'yes' ]; then
   popd
 fi
 
+if [ "${ShouldInstallCK}" == 'yes' ]; then
+  pushd ${CK_BUILD} || exit 1
+
+  time ninja -j ${CKBuildParallelism}
+  if [ $? -ne 0 ]; then
+    exit 1
+  fi
+
+  # TODO: Check parallelism. This may use all available threads.
+  time ninja install
+  if [ $? -ne 0 ]; then
+    exit 1
+  fi
+
+  popd
+fi
+
 # The CK benchmarks repo appears to be private (for the time being).
 
 if [ ! -d ${CK_BENCHMARK_REPO} ]; then