Unverified Commit 9b98f843 authored by Madoura's avatar Madoura
Browse files

rocm-related: create and use a generic updater script

parent 72f5af19

pkgs/development/compilers/hip/default.nix

+19 −2
Original line number Diff line number Diff line
@@ -165,10 +165,19 @@ stdenv.mkDerivation (finalAttrs: { 
    wrapProgram $out/bin/hipconfig --set HIP_PATH $out --set HSA_PATH ${rocm-runtime} --set HIP_CLANG_PATH ${clang}/bin

  '';



  # TODO: Separate HIP and hipamd into separate derivations

  passthru.updateScript = writeScript "update.sh" ''

    #!/usr/bin/env nix-shell

    #!nix-shell -i bash -p curl jq common-updater-scripts nix-prefetch-github

    version="$(curl ''${GITHUB_TOKEN:+"-u \":$GITHUB_TOKEN\""} -sL "https://api.github.com/repos/ROCm-Developer-Tools/HIP/tags" | jq '.[].name | split("-") | .[1] | select( . != null )' --raw-output | sort -n | tail -1)"

    version="$(curl ''${GITHUB_TOKEN:+-u ":$GITHUB_TOKEN"} \

      -sL "https://api.github.com/repos/ROCm-Developer-Tools/HIP/releases?per_page=1" | jq '.[0].tag_name | split("-") | .[1]' --raw-output)"



    IFS='.' read -a version_arr <<< "$version"



    if [ "''${#version_arr[*]}" == 2 ]; then

      version="''${version}.0"

    fi



    current_version="$(grep "version =" pkgs/development/compilers/hip/default.nix | head -n1 | cut -d'"' -f2)"

    if [[ "$version" != "$current_version" ]]; then

      tarball_meta="$(nix-prefetch-github ROCm-Developer-Tools HIP --rev "rocm-$version")"
@@ -180,7 +189,15 @@ stdenv.mkDerivation (finalAttrs: { 
      echo hip already up-to-date

    fi



    version="$(curl ''${GITHUB_TOKEN:+"-u \":$GITHUB_TOKEN\""} -sL "https://api.github.com/repos/ROCm-Developer-Tools/hipamd/tags" | jq '.[].name | split("-") | .[1] | select( . != null )' --raw-output | sort -n | tail -1)"

    version="$(curl ''${GITHUB_TOKEN:+-u ":$GITHUB_TOKEN"} \

      -sL "https://api.github.com/repos/ROCm-Developer-Tools/hipamd/releases?per_page=1" | jq '.[0].tag_name | split("-") | .[1]' --raw-output)"



    IFS='.' read -a version_arr <<< "$version"



    if [ "''${#version_arr[*]}" == 2 ]; then

      version="''${version}.0"

    fi



    current_version="$(grep "version =" pkgs/development/compilers/hip/default.nix | tail -n1 | cut -d'"' -f2)"

    if [[ "$version" != "$current_version" ]]; then

      tarball_meta="$(nix-prefetch-github ROCm-Developer-Tools hipamd --rev "rocm-$version")"

pkgs/development/compilers/llvm/rocm/llvm.nix

+27 −17
Original line number Diff line number Diff line
@@ -58,11 +58,22 @@ in stdenv.mkDerivation (finalAttrs: { 
      --replace 'FILES_MATCHING' 'NO_SOURCE_PERMISSIONS FILES_MATCHING'

  '';



  passthru = {

    isClang = true;



    updateScript = writeScript "update.sh" ''

      #!/usr/bin/env nix-shell

      #!nix-shell -i bash -p curl jq common-updater-scripts nix-prefetch-github



    version="$(curl ''${GITHUB_TOKEN:+"-u \":$GITHUB_TOKEN\""} -sL "https://api.github.com/repos/RadeonOpenCompute/llvm-project/releases?per_page=1" | jq '.[0].tag_name | split("-") | .[1]' --raw-output)"

      version="$(curl ''${GITHUB_TOKEN:+-u ":$GITHUB_TOKEN"} \

        -sL "https://api.github.com/repos/RadeonOpenCompute/llvm-project/releases?per_page=1" | jq '.[0].tag_name | split("-") | .[1]' --raw-output)"



      IFS='.' read -a version_arr <<< "$version"



      if [ "''${#version_arr[*]}" == 2 ]; then

        version="''${version}.0"

      fi



      current_version="$(grep "version =" pkgs/development/compilers/llvm/rocm/default.nix | cut -d'"' -f2)"

      if [[ "$version" != "$current_version" ]]; then

        tarball_meta="$(nix-prefetch-github RadeonOpenCompute llvm-project --rev "rocm-$version")"
@@ -74,8 +85,7 @@ in stdenv.mkDerivation (finalAttrs: { 
        echo rocm-llvm already up-to-date

      fi

    '';



  passthru.isClang = true;

    };



  meta = with lib; {

    description = "ROCm fork of the LLVM compiler infrastructure";

pkgs/development/libraries/clang-ocl/default.nix

+6 −8
Original line number Diff line number Diff line 
{ lib

, stdenv

, fetchFromGitHub

, writeScript

, rocmUpdateScript

, cmake

, rocm-cmake

, rocm-device-libs
@@ -34,13 +34,11 @@ stdenv.mkDerivation (finalAttrs: { 
    "-DCMAKE_CXX_COMPILER=clang++"

  ];



  passthru.updateScript = writeScript "update.sh" ''

    #!/usr/bin/env nix-shell

    #!nix-shell -i bash -p curl jq common-updater-scripts

    version="$(curl ''${GITHUB_TOKEN:+"-u \":$GITHUB_TOKEN\""} \

      -sL "https://api.github.com/repos/RadeonOpenCompute/clang-ocl/releases?per_page=1" | jq '.[0].tag_name | split("-") | .[1]' --raw-output)"

    update-source-version clang-ocl "$version" --ignore-same-hash

  '';

  passthru.updateScript = rocmUpdateScript {

    name = finalAttrs.pname;

    owner = finalAttrs.src.owner;

    repo = finalAttrs.src.repo;

  };



  meta = with lib; {

    description = "OpenCL compilation with clang compiler";

pkgs/development/libraries/hipcub/default.nix

+6 −8
Original line number Diff line number Diff line 
{ lib

, stdenv

, fetchFromGitHub

, writeScript

, rocmUpdateScript

, cmake

, rocm-cmake

, rocm-runtime
@@ -76,13 +76,11 @@ stdenv.mkDerivation (finalAttrs: { 
    rmdir $out/bin

  '';



  passthru.updateScript = writeScript "update.sh" ''

    #!/usr/bin/env nix-shell

    #!nix-shell -i bash -p curl jq common-updater-scripts

    version="$(curl ''${GITHUB_TOKEN:+"-u \":$GITHUB_TOKEN\""} \

      -sL "https://api.github.com/repos/ROCmSoftwarePlatform/hipCUB/releases?per_page=1" | jq '.[0].tag_name | split("-") | .[1]' --raw-output)"

    update-source-version hipcub "$version" --ignore-same-hash

  '';

  passthru.updateScript = rocmUpdateScript {

    name = finalAttrs.pname;

    owner = finalAttrs.src.owner;

    repo = finalAttrs.src.repo;

  };



  meta = with lib; {

    description = "Thin wrapper library on top of rocPRIM or CUB";

pkgs/development/libraries/hipsparse/default.nix

+6 −8
Original line number Diff line number Diff line 
{ lib

, stdenv

, fetchFromGitHub

, writeScript

, rocmUpdateScript

, cmake

, rocm-cmake

, rocm-runtime
@@ -112,13 +112,11 @@ stdenv.mkDerivation (finalAttrs: { 
    rmdir $out/bin

  '';



  passthru.updateScript = writeScript "update.sh" ''

    #!/usr/bin/env nix-shell

    #!nix-shell -i bash -p curl jq common-updater-scripts

    version="$(curl ''${GITHUB_TOKEN:+"-u \":$GITHUB_TOKEN\""} \

      -sL "https://api.github.com/repos/ROCmSoftwarePlatform/hipSPARSE/releases?per_page=1" | jq '.[0].tag_name | split("-") | .[1]' --raw-output)"

    update-source-version hipsparse "$version" --ignore-same-hash

  '';

  passthru.updateScript = rocmUpdateScript {

    name = finalAttrs.pname;

    owner = finalAttrs.src.owner;

    repo = finalAttrs.src.repo;

  };



  meta = with lib; {

    description = "ROCm SPARSE marshalling library";