From 964a8ea439426c82a05098ad52bd982a41124983 Mon Sep 17 00:00:00 2001
From: Brian <bmahabir@bu.edu>
Date: Mon, 20 Jan 2025 17:58:06 -0500
Subject: [PATCH] code crashes for rocm added proper type cast for env var

Signed-off-by: Brian <bmahabir@bu.edu>
---
 ramalama/common.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ramalama/common.py b/ramalama/common.py
index 1d0c33bd..627fe8d1 100644
--- a/ramalama/common.py
+++ b/ramalama/common.py
@@ -230,7 +230,7 @@ def get_gpu():
         i += 1
 
     if gpu_bytes:
-        os.environ["HIP_VISIBLE_DEVICES"] = gpu_num
+        os.environ["HIP_VISIBLE_DEVICES"] = str(gpu_num)
         return