mirror of
https://github.com/kvcache-ai/ktransformers.git
synced 2025-09-05 20:19:51 +00:00
9 lines
339 B
C++
9 lines
339 B
C++
// Adapted from
|
|
// https://github.com/Mozilla-Ocho/llamafile/blob/0.8.8/llamafile/tinyblas_cpu_sgemm_amd_avx512f.cpp
|
|
// Copyrigth 2024 Mozilla Foundation.
|
|
// Copyright(c) 2024 by KVCache.AI, All Rights Reserved.
|
|
|
|
#ifdef __x86_64__
|
|
#define llamafile_sgemm llamafile_sgemm_amd_avx512f
|
|
#include "tinyblas_cpu_sgemm.inc"
|
|
#endif // __x86_64__
|