File tree 2 files changed +16
-21
lines changed
2 files changed +16
-21
lines changed Original file line number Diff line number Diff line change @@ -170,27 +170,23 @@ if (LLAMA_CUBLAS)
170
170
endif ()
171
171
172
172
if (LLAMA_HIPBLAS)
173
- cmake_minimum_required (VERSION 3.21)
174
-
175
173
find_package (hip)
176
174
find_package (hipblas)
177
175
178
- if (hipblas_FOUND)
176
+ if (${ hipblas_FOUND} AND ${hip_FOUND} )
179
177
message (STATUS "hipBLAS found" )
178
+ add_compile_definitions (GGML_USE_HIPBLAS)
179
+ enable_language (HIP)
180
+ add_library (ggml-hip OBJECT ggml-cuda.cu ggml-cuda.h)
181
+ set_source_files_properties (ggml-cuda.cu PROPERTIES LANGUAGE HIP)
182
+ target_link_libraries (ggml-hip PRIVATE hip::device)
180
183
181
- set (LLAMA_HIPBLAS_PLATFORM "AMD" CACHE STRING "hip device type" FORCE)
182
- set_property (CACHE LLAMA_HIPBLAS_PLATFORM PROPERTY STRINGS "AMD" "NVIDIA" )
183
-
184
- add_compile_definitions (GGML_USE_HIPBLAS "__HIP_PLATFORM_${LLAMA_HIPBLAS_PLATFORM} __" )
185
-
186
- add_library (ggml-hip OBJECT ggml-cuda.cu)
187
- set_source_files_properties (ggml-cuda.cu PROPERTIES LANGUAGE CXX)
188
- target_link_libraries (ggml-hip hip::device)
189
-
184
+ if (LLAMA_STATIC)
185
+ message (FATAL_ERROR "Static linking not supported for HIP/ROCm" )
186
+ endif ()
190
187
set (LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS} hip::host roc::hipblas ggml-hip)
191
-
192
188
else ()
193
- message (WARNING "hipBLAS not found" )
189
+ message (WARNING "hipBLAS or HIP not found. Try setting CMAKE_PREFIX_PATH=/opt/rocm " )
194
190
endif ()
195
191
endif ()
196
192
Original file line number Diff line number Diff line change @@ -108,13 +108,12 @@ ggml-cuda.o: ggml-cuda.cu ggml-cuda.h
108
108
nvcc -arch=native -c -o $@ $<
109
109
endif
110
110
ifdef LLAMA_HIPBLAS
111
- ROCMPATH?= /opt/rocm
112
- CFLAGS += -DGGML_USE_HIPBLAS -D__HIP_PLATFORM_AMD__ -I$(ROCMPATH)/include
113
- CXXFLAGS+= -D__HIP_PLATFORM_AMD__ -I$(ROCMPATH)/include
114
- HIPFLAGS?= -amdgpu-early-inline-all=true -amdgpu-function-calls=false -march=native
115
- LDFLAGS += -lhipblas -lamdhip64 -L$(ROCMPATH)/lib
116
- HIPCC ?= $(ROCMPATH)/bin/hipcc
117
- OBJS += ggml-cuda.o
111
+ ROCM_PATH ?= /opt/rocm
112
+ LDFLAGS += -lhipblas -lamdhip64 -L$(ROCM_PATH)/lib
113
+ HIPCC ?= $(ROCM_PATH)/bin/hipcc
114
+ OBJS += ggml-cuda.o
115
+ ggml.o : CFLAGS += -DGGML_USE_HIPBLAS -D__HIP_PLATFORM_AMD__ -I$(ROCM_PATH ) /include
116
+ ggml-cuda.o : CXXFLAGS += -march=native -D__HIP_PLATFORM_AMD__ -I$(ROCMPATH ) /include
118
117
ggml-cuda.o : ggml-cuda.cu ggml-cuda.h
119
118
$(HIPCC ) $(CXXFLAGS ) -x hip $(HIPFLAGS ) -c -o $@ $<
120
119
endif
You can’t perform that action at this time.
0 commit comments