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 <cuda_runtime.h>
#include <cstdint> #include <cstdint>
...@@ -1556,3 +1557,4 @@ infiniStatus_t Descriptor::calculate( ...@@ -1556,3 +1557,4 @@ infiniStatus_t Descriptor::calculate(
} }
} // namespace op::paged_attention_prefill::nvidia } // namespace op::paged_attention_prefill::nvidia
#endif
...@@ -72,7 +72,7 @@ __C infiniStatus_t infiniopCreateSwiGLUDescriptor( ...@@ -72,7 +72,7 @@ __C infiniStatus_t infiniopCreateSwiGLUDescriptor(
CREATE_CUDA(INFINI_DEVICE_ALI, nvidia); CREATE_CUDA(INFINI_DEVICE_ALI, nvidia);
#endif #endif
#ifdef ENABLE_QY_API #ifdef ENABLE_QY_API
CREATE(INFINI_DEVICE_QY, nvidia); CREATE_CUDA(INFINI_DEVICE_QY, nvidia);
#endif #endif
#ifdef ENABLE_HYGON_API #ifdef ENABLE_HYGON_API
CREATE_CUDA(INFINI_DEVICE_HYGON, nvidia); 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