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-: # 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