Skip to content

Commit

Permalink
update makefile and ggml.c
Browse files Browse the repository at this point in the history
  • Loading branch information
YellowRoseCx committed Jul 12, 2023
1 parent 6ca3499 commit 7f85cc5
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 6 deletions.
5 changes: 3 additions & 2 deletions Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -186,8 +186,9 @@ ifdef LLAMA_HIPBLAS
CC := $(ROCM_PATH)/llvm/bin/clang
CXX := $(ROCM_PATH)/llvm/bin/clang++
GPU_TARGETS = gfx803 gfx900 gfx906 gfx908 gfx90a gfx1030 gfx1100
LLAMA_CUDA_DMMV_X ?= 64
LLAMA_CUDA_MMV_Y ?= 2
LLAMA_CUDA_DMMV_X ?= 128
LLAMA_CUDA_MMV_Y ?= 1
LLAMA_CUDA_KQUANTS_ITER ?= 1
LLAMA_CUDA_FORCE_DMMV = true
CFLAGS += -DGGML_USE_HIPBLAS -DGGML_USE_CUBLAS $(shell $(ROCM_PATH)/bin/hipconfig -C)
CXXFLAGS += -DGGML_USE_HIPBLAS -DGGML_USE_CUBLAS $(shell $(ROCM_PATH)/bin/hipconfig -C)
Expand Down
8 changes: 4 additions & 4 deletions ggml.c
Original file line number Diff line number Diff line change
Expand Up @@ -246,12 +246,12 @@ inline static void* ggml_aligned_malloc(size_t size) {
#if defined(GGML_USE_CLBLAST) // allow usage of CLBlast alongside Accelerate functions
#include "ggml-opencl.h"
#endif
#elif defined(GGML_USE_OPENBLAS)
#if defined(GGML_BLAS_USE_MKL)
#include <mkl.h>
#else
#endif
#if defined(GGML_USE_OPENBLAS)
#include <cblas.h>
#endif
#if defined(GGML_BLAS_USE_MKL)
#include <mkl.h>
#endif
#if defined(GGML_USE_CUBLAS)
#include "ggml-cuda.h"
Expand Down

0 comments on commit 7f85cc5

Please sign in to comment.