Skip to content

Commit a719e98

Browse files
Add loongarch compilation support to makefile
1 parent ee42f24 commit a719e98

File tree

2 files changed

+9
-4
lines changed

2 files changed

+9
-4
lines changed

Makefile

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -379,6 +379,11 @@ ifneq ($(filter ppc64le%,$(UNAME_M)),)
379379
CUDA_POWER_ARCH = 1
380380
endif
381381

382+
ifneq ($(filter loongarch64%,$(UNAME_M)),)
383+
MK_CFLAGS += -mlasx
384+
MK_CXXFLAGS += -mlasx
385+
endif
386+
382387
else
383388
MK_CFLAGS += -march=rv64gcv -mabi=lp64d
384389
MK_CXXFLAGS += -march=rv64gcv -mabi=lp64d

ggml-quants.c

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -294,7 +294,7 @@ static __m256 __lasx_xvreplfr2vr_s(float val)
294294
static inline __m256i ____m256i(__m128i in)
295295
{
296296
__m256i out = __lasx_xvldi(0);
297-
asm volatile (
297+
__asm__ volatile (
298298
".irp i," __ALL_REGS "\n\t"
299299
" .ifc %[out], " XREGS_PREFIX"\\i \n\t"
300300
" .irp j," __ALL_REGS "\n\t"
@@ -312,7 +312,7 @@ static inline __m256i ____m256i(__m128i in)
312312
static inline __m256i lasx_set_q(__m128i inhi, __m128i inlo)
313313
{
314314
__m256i out;
315-
asm volatile (
315+
__asm__ volatile (
316316
".irp i," __ALL_REGS "\n\t"
317317
" .ifc %[hi], " VREGS_PREFIX "\\i \n\t"
318318
" .irp j," __ALL_REGS "\n\t"
@@ -342,7 +342,7 @@ static inline __m256i lasx_set_q(__m128i inhi, __m128i inlo)
342342
static inline __m128i lasx_extracti128_lo(__m256i in)
343343
{
344344
__m128i out;
345-
asm volatile (
345+
__asm__ volatile (
346346
".ifnc %[out], %[in] \n\t"
347347
".irp i," __ALL_REGS "\n\t"
348348
" .ifc %[out], " VREGS_PREFIX "\\i \n\t"
@@ -362,7 +362,7 @@ static inline __m128i lasx_extracti128_lo(__m256i in)
362362
static inline __m128i lasx_extracti128_hi(__m256i in)
363363
{
364364
__m128i out;
365-
asm volatile (
365+
__asm__ volatile (
366366
".irp i," __ALL_REGS "\n\t"
367367
" .ifc %[out], " VREGS_PREFIX "\\i \n\t"
368368
" .irp j," __ALL_REGS "\n\t"

0 commit comments

Comments
 (0)