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

src/infiniop/ops/rope/operator.cc

@@ -73,9 +73,9 @@ __C infiniStatus_t infiniopCreateRoPEDescriptor(
 
 __C infiniStatus_t infiniopGetRoPEWorkspaceSize(infiniopRoPEDescriptor_t desc,
                                                 size_t *size) {
-#define GET(CASE, NAMESPACE)                                                                     \
-    case CASE:                                                                                   \
-        *size = reinterpret_cast<const op::rope::NAMESPACE::Descriptor *>(desc)->workspace_size; \
+#define GET(CASE, NAMESPACE)                                                                      \
+    case CASE:                                                                                    \
+        *size = reinterpret_cast<const op::rope::NAMESPACE::Descriptor *>(desc)->workspaceSize(); \
         return INFINI_STATUS_SUCCESS
 
     switch (desc->device_type) {

src/infiniop/ops/rope/rope.h

@@ -5,49 +5,50 @@
 #include "../../operator.h"
 #include "../../tensor.h"
 
-#define DESCRIPTOR(NAMESPACE)                             \
-                                                          \
-    namespace op::rope::NAMESPACE {                       \
-    class Descriptor final : public InfiniopDescriptor {  \
-        struct Opaque;                                    \
-        Opaque *_opaque;                                  \
-        RoPEInfo _info;                                   \
-                                                          \
-        Descriptor(                                       \
-            RoPEInfo info,                                \
-            size_t workspace_size_,                       \
-            Opaque *opaque,                               \
-            infiniDevice_t device_type,                   \
-            int device_id)                                \
-            : InfiniopDescriptor{device_type, device_id}, \
-              _opaque(opaque),                            \
-              _info(info),                                \
-              workspace_size(workspace_size_) {}          \
-                                                          \
-    public:                                               \
-        size_t workspace_size;                            \
-                                                          \
-        ~Descriptor();                                    \
-                                                          \
-        static infiniStatus_t create(                     \
-            infiniopHandle_t handle,                      \
-            Descriptor **desc_ptr,                        \
-            infiniopTensorDescriptor_t y_desc,            \
-            infiniopTensorDescriptor_t x_desc,            \
-            infiniopTensorDescriptor_t pos_desc,          \
-            infiniopTensorDescriptor_t sin_desc,          \
-            infiniopTensorDescriptor_t cos_desc);         \
-                                                          \
-        infiniStatus_t calculate(                         \
-            void *workspace,                              \
-            size_t workspace_size,                        \
-            void *y,                                      \
-            const void *x,                                \
-            const void *pos_ids,                          \
-            const void *sin_table,                        \
-            const void *cos_table,                        \
-            void *stream) const;                          \
-    };                                                    \
+#define DESCRIPTOR(NAMESPACE)                                    \
+                                                                 \
+    namespace op::rope::NAMESPACE {                              \
+    class Descriptor final : public InfiniopDescriptor {         \
+        struct Opaque;                                           \
+        Opaque *_opaque;                                         \
+        RoPEInfo _info;                                          \
+        size_t _workspace_size;                                  \
+                                                                 \
+        Descriptor(                                              \
+            RoPEInfo info,                                       \
+            size_t workspace_size_,                              \
+            Opaque *opaque,                                      \
+            infiniDevice_t device_type,                          \
+            int device_id)                                       \
+            : InfiniopDescriptor{device_type, device_id},        \
+              _opaque(opaque),                                   \
+              _info(info),                                       \
+              _workspace_size(workspace_size_) {}                \
+                                                                 \
+    public:                                                      \
+        ~Descriptor();                                           \
+                                                                 \
+        size_t workspaceSize() const { return _workspace_size; } \
+                                                                 \
+        static infiniStatus_t create(                            \
+            infiniopHandle_t handle,                             \
+            Descriptor **desc_ptr,                               \
+            infiniopTensorDescriptor_t y_desc,                   \
+            infiniopTensorDescriptor_t x_desc,                   \
+            infiniopTensorDescriptor_t pos_desc,                 \
+            infiniopTensorDescriptor_t sin_desc,                 \
+            infiniopTensorDescriptor_t cos_desc);                \
+                                                                 \
+        infiniStatus_t calculate(                                \
+            void *workspace,                                     \
+            size_t workspace_size,                               \
+            void *y,                                             \
+            const void *x,                                       \
+            const void *pos_ids,                                 \
+            const void *sin_table,                               \
+            const void *cos_table,                               \
+            void *stream) const;                                 \
+    };                                                           \
     }
 
 class RoPEInfo {