let package = { writeShellScript , symlinkJoin , clang , coreutils , gzip , bash , rocmPackages }: let rocmtoolkit_joined = symlinkJoin { name = "rocm-merged"; paths = with rocmPackages; [ rocm-core clr rccl miopen miopengemm rocrand rocblas rocsparse hipsparse rocthrust rocprim hipcub roctracer rocfft rocsolver hipfft hipsolver hipblas rocminfo rocm-thunk rocm-comgr rocm-device-libs rocm-runtime clr.icd hipify ]; }; in writeShellScript "run-llamafile" '' if [ $# -eq 0 ] ; then echo "Usage: $0 wizardcoder.llamafile -ngl 9999" exit 2 fi export PATH=${rocmtoolkit_joined}/bin export PATH=$PATH:${clang}/bin export PATH=$PATH:${coreutils}/bin export PATH=$PATH:${gzip}/bin export HIPCC_COMPILE_FLAGS_APPEND="-I${rocmtoolkit_joined}/include -I${rocmtoolkit_joined}/include/rocblas" export HIPCC_LINK_FLAGS_APPEND="-L${rocmtoolkit_joined}/lib" export ROCM_PATH=${rocmtoolkit_joined} export ROCM_SOURCE_DIR=${rocmtoolkit_joined} ulimit -v unlimited ${pkgs.bash}/bin/sh "$@" ''; nixpkgs = fetchTarball "https://github.com/NixOS/nixpkgs/archive/0ef56bec7281e2372338f2dfe7c13327ce96f6bb.tar.gz"; pkgs = import nixpkgs {}; in pkgs.callPackage package {}