Commit 025894f3 authored by PanZezhong's avatar PanZezhong
Browse files

issue/48/fix 将rope info的workspace_size改成私有

parent c905fd63
...@@ -75,7 +75,7 @@ __C infiniStatus_t infiniopGetRoPEWorkspaceSize(infiniopRoPEDescriptor_t desc, ...@@ -75,7 +75,7 @@ __C infiniStatus_t infiniopGetRoPEWorkspaceSize(infiniopRoPEDescriptor_t desc,
size_t *size) { size_t *size) {
#define GET(CASE, NAMESPACE) \ #define GET(CASE, NAMESPACE) \
case CASE: \ case CASE: \
*size = reinterpret_cast<const op::rope::NAMESPACE::Descriptor *>(desc)->workspace_size; \ *size = reinterpret_cast<const op::rope::NAMESPACE::Descriptor *>(desc)->workspaceSize(); \
return INFINI_STATUS_SUCCESS return INFINI_STATUS_SUCCESS
switch (desc->device_type) { switch (desc->device_type) {
......
...@@ -12,6 +12,7 @@ ...@@ -12,6 +12,7 @@
struct Opaque; \ struct Opaque; \
Opaque *_opaque; \ Opaque *_opaque; \
RoPEInfo _info; \ RoPEInfo _info; \
size_t _workspace_size; \
\ \
Descriptor( \ Descriptor( \
RoPEInfo info, \ RoPEInfo info, \
...@@ -22,13 +23,13 @@ ...@@ -22,13 +23,13 @@
: InfiniopDescriptor{device_type, device_id}, \ : InfiniopDescriptor{device_type, device_id}, \
_opaque(opaque), \ _opaque(opaque), \
_info(info), \ _info(info), \
workspace_size(workspace_size_) {} \ _workspace_size(workspace_size_) {} \
\ \
public: \ public: \
size_t workspace_size; \
\
~Descriptor(); \ ~Descriptor(); \
\ \
size_t workspaceSize() const { return _workspace_size; } \
\
static infiniStatus_t create( \ static infiniStatus_t create( \
infiniopHandle_t handle, \ infiniopHandle_t handle, \
Descriptor **desc_ptr, \ Descriptor **desc_ptr, \
......
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