Merge pull request #1064 from InfiniTensor/issue/1031_T1-1-4

【算子比赛2025秋】 T1-1-4

src/infiniop/ops/var/var_desc.h

+#ifndef INFINIOP_VAR_DESCRIPTOR_H_
+#define INFINIOP_VAR_DESCRIPTOR_H_
+#include "../../../utils.h"
+#include "../../operator.h"
+#include "../../tensor.h"
+
+#include "info.h"
+
+#define DESCRIPTOR(NAMESPACE)                                    \
+                                                                 \
+    namespace op::var::NAMESPACE {                               \
+    class Descriptor final : public InfiniopDescriptor {         \
+        struct Opaque;                                           \
+        Opaque *_opaque;                                         \
+        VarInfo _info;                                           \
+        size_t _workspace_size;                                  \
+                                                                 \
+        Descriptor(                                              \
+            Opaque *opaque,                                      \
+            VarInfo info,                                        \
+            size_t workspace_size,                               \
+            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 var_output_desc,          \
+            infiniopTensorDescriptor_t input_desc,               \
+            size_t *dim,                                         \
+            size_t dim_size,                                     \
+            bool unbiased,                                       \
+            bool keepdim);                                       \
+                                                                 \
+        infiniStatus_t calculate(                                \
+            void *workspace, size_t workspace_size,              \
+            void *var_output,                                    \
+            const void *input,                                   \
+            bool unbiased,                                       \
+            bool keepdim,                                        \
+            void *stream) const;                                 \
+    };                                                           \
+    }
+
+#endif

src/infiniop/ops/var_mean/cpu/var_mean_cpu.h

+#ifndef __INFINIOP_VAR_MEAN_CPU_H__
+#define __INFINIOP_VAR_MEAN_CPU_H__
+
+#include "../var_mean_desc.h"
+
+DESCRIPTOR(cpu);
+
+#endif // __INFINIOP_VAR_MEAN_CPU_H__