From 559902b8372a9fe92805681dc3a002e4870ef3ee Mon Sep 17 00:00:00 2001
From: Kyle Gospodnetich <me@kylegospodneti.ch>
Date: Thu, 6 Jul 2023 15:02:56 -0700
Subject: [PATCH] feat: Add ROCM-HIP and ROCM-OpenCL

---
 Containerfile | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/Containerfile b/Containerfile
index 1aae7aa5..1fe209c5 100644
--- a/Containerfile
+++ b/Containerfile
@@ -54,6 +54,13 @@ RUN rpm-ostree override remove \
     toolbox \
     htop
 
+# Install ROCM on non-Nvidia images
+RUN if grep -v "nvidia" <<< "${IMAGE_NAME}"; then \
+    rpm-ostree install \
+        rocm-hip \
+        rocm-opencl \
+; fi 
+
 # Run firstboot script per-profile
 RUN mkdir -p "/usr/etc/profile.d/"
 RUN ln -s "/usr/share/ublue-os/firstboot/launcher/login-profile.sh" \