Skip to content

Commit c236e57

Browse files
Add loongarch compilation support to makefile
1 parent 1a2a17b commit c236e57

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
@@ -351,6 +351,11 @@ ifneq ($(filter ppc64le%,$(UNAME_M)),)
351351
CUDA_POWER_ARCH = 1
352352
endif
353353

354+
ifneq ($(filter loongarch64%,$(UNAME_M)),)
355+
MK_CFLAGS += -mlasx
356+
MK_CXXFLAGS += -mlasx
357+
endif
358+
354359
else
355360
MK_CFLAGS += -march=rv64gcv -mabi=lp64d
356361
MK_CXXFLAGS += -march=rv64gcv -mabi=lp64d

ggml-quants.c

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -584,7 +584,7 @@ static __m256 __lasx_xvreplfr2vr_s(float val)
584584
static inline __m256i ____m256i(__m128i in)
585585
{
586586
__m256i out = __lasx_xvldi(0);
587-
asm volatile (
587+
__asm__ volatile (
588588
".irp i," __ALL_REGS "\n\t"
589589
" .ifc %[out], " XREGS_PREFIX"\\i \n\t"
590590
" .irp j," __ALL_REGS "\n\t"
@@ -602,7 +602,7 @@ static inline __m256i ____m256i(__m128i in)
602602
static inline __m256i lasx_set_q(__m128i inhi, __m128i inlo)
603603
{
604604
__m256i out;
605-
asm volatile (
605+
__asm__ volatile (
606606
".irp i," __ALL_REGS "\n\t"
607607
" .ifc %[hi], " VREGS_PREFIX "\\i \n\t"
608608
" .irp j," __ALL_REGS "\n\t"
@@ -632,7 +632,7 @@ static inline __m256i lasx_set_q(__m128i inhi, __m128i inlo)
632632
static inline __m128i lasx_extracti128_lo(__m256i in)
633633
{
634634
__m128i out;
635-
asm volatile (
635+
__asm__ volatile (
636636
".ifnc %[out], %[in] \n\t"
637637
".irp i," __ALL_REGS "\n\t"
638638
" .ifc %[out], " VREGS_PREFIX "\\i \n\t"
@@ -652,7 +652,7 @@ static inline __m128i lasx_extracti128_lo(__m256i in)
652652
static inline __m128i lasx_extracti128_hi(__m256i in)
653653
{
654654
__m128i out;
655-
asm volatile (
655+
__asm__ volatile (
656656
".irp i," __ALL_REGS "\n\t"
657657
" .ifc %[out], " VREGS_PREFIX "\\i \n\t"
658658
" .irp j," __ALL_REGS "\n\t"

0 commit comments

Comments
 (0)