Commit 832e607e authored by xgqdut2016's avatar xgqdut2016
Browse files

issue/1048: debug qy swiglu and paged_attention_prefill

parent a9503148
#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