mirror of
https://github.com/kvcache-ai/ktransformers.git
synced 2025-09-05 12:09:48 +00:00
9 lines
360 B
C++
9 lines
360 B
C++
// Adapted from
|
|
// https://github.com/Mozilla-Ocho/llamafile/blob/0.8.8/llamafile/tinyblas_cpu_mixmul_amd_fma.cpp
|
|
// Copyrigth 2024 Mozilla Foundation.
|
|
// Copyright(c) 2024 by KVCache.AI, All Rights Reserved.
|
|
|
|
#if defined(__x86_64__) || defined(_M_X64)
|
|
#define llamafile_mixmul llamafile_mixmul_amd_fma
|
|
#include "tinyblas_cpu_mixmul.inc"
|
|
#endif // __x86_64__
|