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 <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