Unverified Commit 3e5cad10 authored by thatPepe's avatar thatPepe Committed by GitHub
Browse files

Merge pull request #1049 from InfiniTensor/issue/1048

issue/1048: 修复QY机器上的paged_attention_prefill和swiglu编译问题
parents 69a74ac5 832e607e
#if defined(ENABLE_NVIDIA_API) || defined(ENABLE_ALI_API) || defined(ENABLE_ILUVATAR_API)
#include <cuda_runtime.h>
#include <cstdint>
......@@ -1556,3 +1557,4 @@ infiniStatus_t Descriptor::calculate(
}
} // namespace op::paged_attention_prefill::nvidia
#endif
......@@ -72,7 +72,7 @@ __C infiniStatus_t infiniopCreateSwiGLUDescriptor(
CREATE_CUDA(INFINI_DEVICE_ALI, nvidia);
#endif
#ifdef ENABLE_QY_API
CREATE(INFINI_DEVICE_QY, nvidia);
CREATE_CUDA(INFINI_DEVICE_QY, nvidia);
#endif
#ifdef ENABLE_HYGON_API
CREATE_CUDA(INFINI_DEVICE_HYGON, nvidia);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment