Commit 1443aa67 authored by wooway777's avatar wooway777
Browse files

issue/1032 - adjust swiglu operator.cc

parent ae49716d
...@@ -102,6 +102,7 @@ __C infiniStatus_t infiniopCreateSwiGLUDescriptor( ...@@ -102,6 +102,7 @@ __C infiniStatus_t infiniopCreateSwiGLUDescriptor(
} }
#undef CREATE #undef CREATE
#undef CREATE_CUDA
} }
__C infiniStatus_t infiniopGetSwiGLUWorkspaceSize(infiniopSwiGLUDescriptor_t desc, size_t *size) { __C infiniStatus_t infiniopGetSwiGLUWorkspaceSize(infiniopSwiGLUDescriptor_t desc, size_t *size) {
...@@ -159,11 +160,12 @@ __C infiniStatus_t infiniopGetSwiGLUWorkspaceSize(infiniopSwiGLUDescriptor_t des ...@@ -159,11 +160,12 @@ __C infiniStatus_t infiniopGetSwiGLUWorkspaceSize(infiniopSwiGLUDescriptor_t des
#ifdef ENABLE_MOORE_API #ifdef ENABLE_MOORE_API
GET(INFINI_DEVICE_MOORE, moore); GET(INFINI_DEVICE_MOORE, moore);
#endif #endif
default:
return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;
} }
#undef GET #undef GET
#undef GET_CUDA
return INFINI_STATUS_DEVICE_TYPE_NOT_SUPPORTED;
} }
__C infiniStatus_t infiniopSwiGLU( __C infiniStatus_t infiniopSwiGLU(
...@@ -234,6 +236,7 @@ __C infiniStatus_t infiniopSwiGLU( ...@@ -234,6 +236,7 @@ __C infiniStatus_t infiniopSwiGLU(
} }
#undef CALCULATE #undef CALCULATE
#undef CALCULATE_CUDA
} }
__C infiniStatus_t __C infiniStatus_t
...@@ -298,4 +301,5 @@ infiniopDestroySwiGLUDescriptor(infiniopSwiGLUDescriptor_t desc) { ...@@ -298,4 +301,5 @@ infiniopDestroySwiGLUDescriptor(infiniopSwiGLUDescriptor_t desc) {
} }
#undef DELETE #undef DELETE
#undef DELETE_CUDA
} }
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