issue/50/refactor: workspace_size 改为 minWorkSpaceSize()

Signed-off-by: YdrMaster <ydrml@hotmail.com>

src/infiniop/ops/random_sample/cpu/random_sample_cpu.cc

@@ -41,6 +41,10 @@ infiniStatus_t Descriptor::create(
     return INFINI_STATUS_SUCCESS;
 }
 
+size_t Descriptor::minWorkspaceSize() const {
+    return _min_workspace_size;
+}
+
 template <typename DT>
 struct ComputeType {
     using type = DT;

src/infiniop/ops/random_sample/operator.cc

@@ -37,9 +37,10 @@ __C infiniStatus_t infiniopGetRandomSampleWorkspaceSize(
     infiniopRandomSampleDescriptor_t desc,
     size_t *size) {
 
-#define GET(CASE, NAMESPACE)                                                                              \
-    case CASE:                                                                                            \
-        *size = reinterpret_cast<const op::random_sample::NAMESPACE::Descriptor *>(desc)->workspace_size; \
+#define GET(CASE, NAMESPACE)                                          \
+    case CASE:                                                        \
+        using Ptr = const op::random_sample::NAMESPACE::Descriptor *; \
+        *size = reinterpret_cast<Ptr>(desc)->minWorkspaceSize();      \
         return INFINI_STATUS_SUCCESS
 
     switch (desc->device_type) {

src/infiniop/ops/random_sample/random_sample.h

@@ -12,13 +12,13 @@
         Opaque *_opaque;                                  \
                                                           \
         infiniDtype_t _dt_i, _dt_p;                       \
-        size_t _n;                                        \
+        size_t _n, _min_workspace_size;                   \
                                                           \
         Descriptor(                                       \
             infiniDtype_t dt_i,                           \
             infiniDtype_t dt_p,                           \
             size_t n,                                     \
-            size_t workspace_size_,                       \
+            size_t min_workspace_size,                    \
             Opaque *opaque,                               \
             infiniDevice_t device_type,                   \
             int device_id)                                \
@@ -27,11 +27,9 @@
               _dt_i(dt_i),                                \
               _dt_p(dt_p),                                \
               _n(n),                                      \
-              workspace_size(workspace_size_) {}          \
+              _min_workspace_size(min_workspace_size) {}  \
                                                           \
     public:                                               \
-        size_t workspace_size;                            \
-                                                          \
         ~Descriptor();                                    \
                                                           \
         static infiniStatus_t create(                     \
@@ -40,6 +38,8 @@
             infiniopTensorDescriptor_t result_desc,       \
             infiniopTensorDescriptor_t probs_desc);       \
                                                           \
+        size_t minWorkspaceSize() const;                  \
+                                                          \
         infiniStatus_t calculate(                         \
             void *workspace,                              \
             size_t workspace_size,                        \