mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-14 10:09:41 +00:00
98 lines
2.5 KiB
Bash
98 lines
2.5 KiB
Bash
#!/bin/bash
|
|
# Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
# Common utilities for CK Docker tools
|
|
# Shared configuration and helper functions
|
|
|
|
# Find project root (where .git directory is)
|
|
get_project_root() {
|
|
local script_dir="$1"
|
|
cd "${script_dir}/../.." && pwd
|
|
}
|
|
|
|
# Detect git branch and sanitize for Docker naming
|
|
get_sanitized_branch() {
|
|
local project_root="$1"
|
|
local branch
|
|
|
|
branch=$(cd "${project_root}" && git rev-parse --abbrev-ref HEAD 2>/dev/null | tr '/' '_' | tr -cd 'a-zA-Z0-9_-' || echo "")
|
|
branch=${branch:-unknown}
|
|
|
|
# Handle detached HEAD state
|
|
if [ "${branch}" = "HEAD" ]; then
|
|
branch="detached"
|
|
fi
|
|
|
|
echo "${branch}"
|
|
}
|
|
|
|
# Get username with fallback
|
|
get_username() {
|
|
echo "${USER:-$(whoami 2>/dev/null || echo "user")}"
|
|
}
|
|
|
|
# Generate default container name: ck_<username>_<branch>
|
|
get_default_container_name() {
|
|
local project_root="$1"
|
|
local user_name
|
|
local git_branch
|
|
|
|
user_name=$(get_username)
|
|
git_branch=$(get_sanitized_branch "${project_root}")
|
|
|
|
echo "ck_${user_name}_${git_branch}"
|
|
}
|
|
|
|
# Get container name (respects CK_CONTAINER_NAME env var)
|
|
get_container_name() {
|
|
local project_root="$1"
|
|
local default_name
|
|
|
|
default_name=$(get_default_container_name "${project_root}")
|
|
echo "${CK_CONTAINER_NAME:-${default_name}}"
|
|
}
|
|
|
|
# Get Docker image (respects CK_DOCKER_IMAGE env var)
|
|
get_docker_image() {
|
|
echo "${CK_DOCKER_IMAGE:-rocm/composable_kernel:ck_ub24.04_rocm7.0.1}"
|
|
}
|
|
|
|
# Check if container exists (exact match)
|
|
container_exists() {
|
|
local name="$1"
|
|
docker ps -a --filter "name=^${name}$" --format '{{.Names}}' | grep -q "^${name}$"
|
|
}
|
|
|
|
# Check if container is running (exact match)
|
|
container_is_running() {
|
|
local name="$1"
|
|
docker ps --filter "name=^${name}$" --format '{{.Names}}' | grep -q "^${name}$"
|
|
}
|
|
|
|
# Detect GPU target in container
|
|
detect_gpu_target() {
|
|
local container="$1"
|
|
|
|
# Allow override via GPU_TARGET environment variable
|
|
if [ -n "${GPU_TARGET:-}" ]; then
|
|
echo "${GPU_TARGET}"
|
|
return 0
|
|
fi
|
|
|
|
docker exec "${container}" bash -c "
|
|
rocminfo 2>/dev/null | grep -oP 'gfx[0-9a-z]+' | head -1 || echo 'gfx950'
|
|
" | tr -d '\r\n'
|
|
}
|
|
|
|
# Ensure container is running, start if needed
|
|
ensure_container_running() {
|
|
local container="$1"
|
|
local script_dir="$2"
|
|
|
|
if ! container_is_running "${container}"; then
|
|
echo "Container '${container}' not running. Starting with ck-docker..."
|
|
"${script_dir}/ck-docker" start "${container}"
|
|
fi
|
|
}
|