From 04ce04ae966a33c483196a6d5b28d3113cf75966 Mon Sep 17 00:00:00 2001
From: MatheusGASource <matheusgasource@gmail.com>
Date: Fri, 15 Dec 2023 13:57:47 -0300
Subject: [PATCH] build : Check the ROCm installation location

---
 Makefile | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/Makefile b/Makefile
index fb775ae5b..c127046f6 100644
--- a/Makefile
+++ b/Makefile
@@ -439,9 +439,17 @@ ggml-opencl.o: ggml-opencl.cpp ggml-opencl.h
 endif # LLAMA_CLBLAST
 
 ifdef LLAMA_HIPBLAS
-	ROCM_PATH	?= /opt/rocm
-	HIPCC	    ?= $(ROCM_PATH)/bin/hipcc
-	GPU_TARGETS ?= $(shell $(ROCM_PATH)/llvm/bin/amdgpu-arch)
+
+	ifeq ($(wildcard /opt/rocm),)
+		ROCM_PATH	?= /usr
+		HIPCC	    ?= $(ROCM_PATH)/bin/hipcc
+		GPU_TARGETS ?= $(shell $(ROCM_PATH)/bin/amdgpu-arch)
+	else
+		ROCM_PATH	?= /opt/rocm
+		HIPCC	    ?= $(ROCM_PATH)/bin/hipcc
+		GPU_TARGETS ?= $(shell $(ROCM_PATH)/llvm/bin/amdgpu-arch)
+	endif
+	
 	LLAMA_CUDA_DMMV_X       ?= 32
 	LLAMA_CUDA_MMV_Y        ?= 1
 	LLAMA_CUDA_KQUANTS_ITER ?= 2