issue/1031 merge T1-1-2

Author: LaiQuan-conquer

include/infiniop.h

@@ -19,6 +19,7 @@
 #include "infiniop/ops/attention.h"
 #include "infiniop/ops/avg_pool1d.h"
 #include "infiniop/ops/binary_cross_entropy_with_logits.h"
+#include "infiniop/ops/avg_pool3d.h"
 #include "infiniop/ops/causal_softmax.h"
 #include "infiniop/ops/cdist.h"
 #include "infiniop/ops/clip.h"
@@ -27,6 +28,7 @@
 #include "infiniop/ops/dequant/per_tensor_dequant_int8.h"
 #include "infiniop/ops/dequantize_awq.h"
 #include "infiniop/ops/dequantize_gptq.h"
+#include "infiniop/ops/dot.h"
 #include "infiniop/ops/embedding.h"
 #include "infiniop/ops/equal.h"
 #include "infiniop/ops/flash_attention.h"
@@ -44,12 +46,15 @@
 #include "infiniop/ops/index_add.h"
 #include "infiniop/ops/index_copy.h"
 #include "infiniop/ops/inner.h"
+#include "infiniop/ops/histc.h"
 #include "infiniop/ops/int8_gemm.h"
 #include "infiniop/ops/kthvalue.h"
 #include "infiniop/ops/kv_caching.h"
 #include "infiniop/ops/layer_norm.h"
 #include "infiniop/ops/ldexp.h"
 #include "infiniop/ops/lerp.h"
+#include "infiniop/ops/log10.h"
+#include "infiniop/ops/log1p.h"
 #include "infiniop/ops/logsoftmax.h"
 #include "infiniop/ops/lp_norm.h"
 #include "infiniop/ops/masked_select.h"

include/infiniop/ops/avg_pool3d.h

@@ -0,0 +1,27 @@
+#ifndef __INFINIOP_AVG_POOL3D_API_H__
+#define __INFINIOP_AVG_POOL3D_API_H__
+
+#include "../operator_descriptor.h"
+
+typedef struct InfiniopDescriptor *infiniopAvgPool3dDescriptor_t;
+
+__C __export infiniStatus_t infiniopCreateAvgPool3dDescriptor(infiniopHandle_t handle,
+                                                               infiniopAvgPool3dDescriptor_t *desc_ptr,
+                                                               infiniopTensorDescriptor_t y,
+                                                               infiniopTensorDescriptor_t x,
+                                                               void *kernel_size,
+                                                               void *stride,
+                                                               void *padding);
+
+__C __export infiniStatus_t infiniopGetAvgPool3dWorkspaceSize(infiniopAvgPool3dDescriptor_t desc, size_t *size);
+
+__C __export infiniStatus_t infiniopAvgPool3d(infiniopAvgPool3dDescriptor_t desc,
+                                               void *workspace,
+                                               size_t workspace_size,
+                                               void *y,
+                                               const void *x,
+                                               void *stream);
+
+__C __export infiniStatus_t infiniopDestroyAvgPool3dDescriptor(infiniopAvgPool3dDescriptor_t desc);
+
+#endif

include/infiniop/ops/dot.h

@@ -0,0 +1,26 @@
+#ifndef __INFINIOP_DOT_API_H__
+#define __INFINIOP_DOT_API_H__
+
+#include "../operator_descriptor.h"
+
+typedef struct InfiniopDescriptor *infiniopDotDescriptor_t;
+
+__C __export infiniStatus_t infiniopCreateDotDescriptor(infiniopHandle_t handle,
+                                                        infiniopDotDescriptor_t *desc_ptr,
+                                                        infiniopTensorDescriptor_t y,
+                                                        infiniopTensorDescriptor_t a,
+                                                        infiniopTensorDescriptor_t b);
+
+__C __export infiniStatus_t infiniopGetDotWorkspaceSize(infiniopDotDescriptor_t desc, size_t *size);
+
+__C __export infiniStatus_t infiniopDot(infiniopDotDescriptor_t desc,
+                                        void *workspace,
+                                        size_t workspace_size,
+                                        void *y,
+                                        const void *a,
+                                        const void *b,
+                                        void *stream);
+
+__C __export infiniStatus_t infiniopDestroyDotDescriptor(infiniopDotDescriptor_t desc);
+
+#endif

include/infiniop/ops/histc.h

@@ -0,0 +1,28 @@
+#ifndef __INFINIOP_HISTC_API_H__
+#define __INFINIOP_HISTC_API_H__
+
+#include "../operator_descriptor.h"
+#include <stdint.h>
+
+typedef struct InfiniopDescriptor *infiniopHistcDescriptor_t;
+
+__C __export infiniStatus_t infiniopCreateHistcDescriptor(infiniopHandle_t handle,
+                                                           infiniopHistcDescriptor_t *desc_ptr,
+                                                           infiniopTensorDescriptor_t y,
+                                                           infiniopTensorDescriptor_t x,
+                                                           int64_t bins,
+                                                           double min_val,
+                                                           double max_val);
+
+__C __export infiniStatus_t infiniopGetHistcWorkspaceSize(infiniopHistcDescriptor_t desc, size_t *size);
+
+__C __export infiniStatus_t infiniopHistc(infiniopHistcDescriptor_t desc,
+                                           void *workspace,
+                                           size_t workspace_size,
+                                           void *y,
+                                           const void *x,
+                                           void *stream);
+
+__C __export infiniStatus_t infiniopDestroyHistcDescriptor(infiniopHistcDescriptor_t desc);
+
+#endif

include/infiniop/ops/log10.h

@@ -0,0 +1,24 @@
+#ifndef __INFINIOP_LOG10_API_H__
+#define __INFINIOP_LOG10_API_H__
+
+#include "../operator_descriptor.h"
+
+typedef struct InfiniopDescriptor *infiniopLog10Descriptor_t;
+
+__C __export infiniStatus_t infiniopCreateLog10Descriptor(infiniopHandle_t handle,
+                                                          infiniopLog10Descriptor_t *desc_ptr,
+                                                          infiniopTensorDescriptor_t y,
+                                                          infiniopTensorDescriptor_t x);
+
+__C __export infiniStatus_t infiniopGetLog10WorkspaceSize(infiniopLog10Descriptor_t desc, size_t *size);
+
+__C __export infiniStatus_t infiniopLog10(infiniopLog10Descriptor_t desc,
+                                          void *workspace,
+                                          size_t workspace_size,
+                                          void *y,
+                                          const void *x,
+                                          void *stream);
+
+__C __export infiniStatus_t infiniopDestroyLog10Descriptor(infiniopLog10Descriptor_t desc);
+
+#endif

include/infiniop/ops/log1p.h

@@ -0,0 +1,24 @@
+#ifndef __INFINIOP_LOG1P_API_H__
+#define __INFINIOP_LOG1P_API_H__
+
+#include "../operator_descriptor.h"
+
+typedef struct InfiniopDescriptor *infiniopLog1pDescriptor_t;
+
+__C __export infiniStatus_t infiniopCreateLog1pDescriptor(infiniopHandle_t handle,
+                                                          infiniopLog1pDescriptor_t *desc_ptr,
+                                                          infiniopTensorDescriptor_t y,
+                                                          infiniopTensorDescriptor_t x);
+
+__C __export infiniStatus_t infiniopGetLog1pWorkspaceSize(infiniopLog1pDescriptor_t desc, size_t *size);
+
+__C __export infiniStatus_t infiniopLog1p(infiniopLog1pDescriptor_t desc,
+                                          void *workspace,
+                                          size_t workspace_size,
+                                          void *y,
+                                          const void *x,
+                                          void *stream);
+
+__C __export infiniStatus_t infiniopDestroyLog1pDescriptor(infiniopLog1pDescriptor_t desc);
+
+#endif