mirror of
https://github.com/kvcache-ai/ktransformers.git
synced 2026-04-28 03:39:48 +00:00
e69c67713f
Some checks failed
Book-CI / test (push) Has been cancelled
Book-CI / test-1 (push) Has been cancelled
Book-CI / test-2 (push) Has been cancelled
Deploy / deploy (macos-latest) (push) Has been cancelled
Deploy / deploy (ubuntu-latest) (push) Has been cancelled
Deploy / deploy (windows-latest) (push) Has been cancelled
* [refactor]: relocate third_party directory * [fix]: fix custom_flashinfer for kt-sft
9 lines
362 B
C++
9 lines
362 B
C++
// Adapted from
|
|
// https://github.com/Mozilla-Ocho/llamafile/blob/0.8.8/llamafile/tinyblas_cpu_mixmul_amd_zen4.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_zen4
|
|
#include "tinyblas_cpu_mixmul.inc"
|
|
#endif // __x86_64__
|