Includes patch suggested by @shuni64 which fixes half precision ABI issues Includes hipblaslt compression patch Includes configurable hipblaslt support in rocblas rocmPackages_6.hipblaslt: respect NIX_BUILD_CORES in tensilelite rocmPackages_6.hipblas: propagate hipblas-common rocmPackages_6.clr: avoid confusion with hipClangPath Co-authored-by: Gavin Zhao <git@gzgz.dev>
63 lines
1.9 KiB
Nix
63 lines
1.9 KiB
Nix
{
|
|
lib,
|
|
writeScript,
|
|
}:
|
|
|
|
{
|
|
name ? "",
|
|
owner ? "",
|
|
repo ? "",
|
|
page ? "releases",
|
|
# input: array of [ { tag_name: "rocm-6.x.x", }, ... ]. some entries may have bad names like rocm-test-date we want to skip
|
|
# output: first tag_name/name that's a proper version if any
|
|
filter ? "map(.tag_name // .name) | map(select(test(\"^rocm-[0-9]+\\\\.[0-9]+(\\\\.[0-9]+)?$\"))) | first | ltrimstr(\"rocm-\")",
|
|
}:
|
|
|
|
let
|
|
pname =
|
|
if lib.hasPrefix "rocm-llvm-" name then "llvm.${lib.removePrefix "rocm-llvm-" name}" else name;
|
|
|
|
updateScript = writeScript "update.sh" ''
|
|
#!/usr/bin/env nix-shell
|
|
#!nix-shell -i bash -p curl jq common-updater-scripts
|
|
set -euo pipefail
|
|
|
|
fetch_releases() {
|
|
local api_url="https://api.github.com/repos/${owner}/${repo}/${page}"
|
|
if [ "${page}" = "releases" ]; then
|
|
api_url="$api_url?per_page=4"
|
|
fi
|
|
>&2 echo $api_url
|
|
curl ''${GITHUB_TOKEN:+-u ":$GITHUB_TOKEN"} -sL "$api_url"
|
|
}
|
|
|
|
find_valid_version() {
|
|
local releases="$1"
|
|
>&2 echo "$releases"
|
|
# Wrap in array if not already an array to make handline specific release or tags page the same
|
|
>&2 echo jq -r 'if type == "array" then . else [.] end | ${filter}'
|
|
echo "$releases" | jq -r 'if type == "array" then . else [.] end | ${filter}'
|
|
}
|
|
|
|
releases="$(fetch_releases)"
|
|
version="$(find_valid_version "$releases")"
|
|
|
|
if [ -z "$version" ]; then
|
|
echo "No valid version found in the fetched release(s)." >&2
|
|
exit 1
|
|
fi
|
|
|
|
IFS='.' read -ra version_arr <<< "$version"
|
|
|
|
>&2 echo parsed version "$version_arr" from "$version"
|
|
|
|
if (( ''${version_arr[0]} > 6 )); then
|
|
echo "'rocmPackages_6.${pname}' is already at its maximum allowed version.''\nAny further upgrades should go into 'rocmPackages_X.${pname}'." >&2
|
|
exit 1
|
|
fi
|
|
|
|
update-source-version rocmPackages_6.${pname} "$version" --ignore-same-hash
|
|
'';
|
|
in
|
|
[ updateScript ]
|