From f152b80b72152419c34c606bfa32d90fa6b78852 Mon Sep 17 00:00:00 2001
From: Keita Watanabe <mlkeita@amazon.com>
Date: Mon, 21 Oct 2024 18:30:02 +0900
Subject: [PATCH] add GPU accounting

---
 .../LifecycleScripts/base-config/setup_mariadb_accounting.sh   | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/1.architectures/5.sagemaker-hyperpod/LifecycleScripts/base-config/setup_mariadb_accounting.sh b/1.architectures/5.sagemaker-hyperpod/LifecycleScripts/base-config/setup_mariadb_accounting.sh
index 8dcfb622..0694b5dc 100644
--- a/1.architectures/5.sagemaker-hyperpod/LifecycleScripts/base-config/setup_mariadb_accounting.sh
+++ b/1.architectures/5.sagemaker-hyperpod/LifecycleScripts/base-config/setup_mariadb_accounting.sh
@@ -99,6 +99,9 @@ JobAcctGatherFrequency=30
 AccountingStorageType=accounting_storage/slurmdbd
 AccountingStorageHost=$DBD_HOST
 AccountingStoragePort=6819
+AccountingStorageUser=slurm
+AccountingStorageTRES=gres/gpu
+GresTypes=gpu
 EOL
 }