From bef08b17cc0018b7426469b5fff49ab3b8bc255b Mon Sep 17 00:00:00 2001 From: "C.J. Collier" Date: Thu, 9 Jan 2025 19:02:43 -0800 Subject: [PATCH] migrated rapids.sh base template to rapids-template-20250106 --- templates/rapids/rapids.sh.in | 63 ----------------------------------- 1 file changed, 63 deletions(-) delete mode 100644 templates/rapids/rapids.sh.in diff --git a/templates/rapids/rapids.sh.in b/templates/rapids/rapids.sh.in deleted file mode 100644 index 61b7247c0..000000000 --- a/templates/rapids/rapids.sh.in +++ /dev/null @@ -1,63 +0,0 @@ -#!/bin/bash -# -[% INSERT legal/license_header %] -# -[% PROCESS common/template_disclaimer %] -# -# This initialization action script will install rapids on a Dataproc -# cluster. - -set -euxo pipefail - -[% INSERT common/util_functions %] - -[% INSERT gpu/util_functions %] - -[% INSERT dask/util_functions %] - -function main() { - # Install Dask with RAPIDS - install_dask_rapids - - # In "standalone" mode, Dask relies on a systemd unit to launch. - # In "yarn" mode, it relies a config.yaml file. - if [[ "${DASK_RUNTIME}" == "yarn" ]]; then - # Create cuda accelerated Dask YARN config file - configure_dask_yarn - echo "yarn setup complete" - else - # Create Dask service - install_systemd_dask_service - start_systemd_dask_service - - configure_knox_for_dask - - local DASK_CLOUD_LOGGING="$(get_metadata_attribute dask-cloud-logging 'false')" - if [[ "${DASK_CLOUD_LOGGING}" == "true" ]]; then - configure_fluentd_for_dask - fi - fi -} - -function exit_handler() { - gpu_exit_handler - pip_exit_handler - conda_exit_handler - common_exit_handler - return 0 -} - -function prepare_to_install(){ - prepare_common_env - conda_env="$(get_metadata_attribute conda-env 'dask-rapids')" - readonly conda_env - prepare_dask_rapids_env - prepare_conda_env - prepare_pip_env - prepare_gpu_env - trap exit_handler EXIT -} - -prepare_to_install - -main