From 56b91ff9081bebce8edd30ec2f0b11668b59ceff Mon Sep 17 00:00:00 2001
From: Vignesh Edithal <Vignesh.Edithal@amd.com>
Date: Thu, 25 Sep 2025 10:17:02 -0400
Subject: [PATCH] Update docker compose file

* Forward ROCR_VISIBLE_DEVICES to use reserved GPUs

* Write to a consistent image name upon build

* Add username placeholder to service name field to prevent container name clashes
---
 .../docker/docker-compose.customrocmtest.yml                | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/projects/rocprofiler-compute/docker/docker-compose.customrocmtest.yml b/projects/rocprofiler-compute/docker/docker-compose.customrocmtest.yml
index 8b2ce9c31ab9..efce3257d8b1 100644
--- a/projects/rocprofiler-compute/docker/docker-compose.customrocmtest.yml
+++ b/projects/rocprofiler-compute/docker/docker-compose.customrocmtest.yml
@@ -1,8 +1,12 @@
 services:
-  customrocmtest: # service name
+  rocprof-compute-<username>: # service name
     build:
       context: ../../../
       dockerfile: projects/rocprofiler-compute/docker/Dockerfile.customrocmtest
+    image: docker-customrocmtest:latest
+    # Forward some env vars
+    environment:
+      ROCR_VISIBLE_DEVICES: ${ROCR_VISIBLE_DEVICES}
     devices:
       - /dev/kfd
       - /dev/dri