issue/1031 merge T1-1-5

Author: lengmuzhaxi

include/infinicore/ops/acos.hpp

@@ -0,0 +1,16 @@
+#pragma once
+
+#include "../device.hpp"
+#include "common/op.hpp"
+
+namespace infinicore::op {
+class Acos {
+public:
+    using schema = void (*)(Tensor, Tensor);
+    static void execute(Tensor output, Tensor input);
+    static common::OpDispatcher<schema> &dispatcher();
+};
+
+Tensor acos(Tensor input);
+void acos_(Tensor output, Tensor input);
+} // namespace infinicore::op

include/infinicore/ops/adaptive_avg_pool1d.hpp

@@ -0,0 +1,18 @@
+#pragma once
+
+#include "../device.hpp"
+#include "common/op.hpp"
+
+namespace infinicore::op {
+
+class AdaptiveAvgPool1d {
+public:
+    // Schema: execute(Output, Input)
+    using schema = void (*)(Tensor, Tensor);
+    static void execute(Tensor output, Tensor input);
+    static common::OpDispatcher<schema> &dispatcher();
+};
+
+Tensor adaptive_avg_pool1d(Tensor input, int64_t output_size);
+
+} // namespace infinicore::op

include/infinicore/ops/addbmm.hpp

@@ -0,0 +1,19 @@
+#pragma once
+
+#include "../device.hpp"
+#include "common/op.hpp"
+
+namespace infinicore::op {
+
+class Addbmm {
+public:
+    using schema = void (*)(Tensor, Tensor, Tensor, Tensor, float, float);
+    static void execute(Tensor output, Tensor input, Tensor batch1, Tensor batch2, float beta, float alpha);
+    
+    static common::OpDispatcher<schema> &dispatcher();
+};
+Tensor addbmm(Tensor input, Tensor batch1, Tensor batch2, float beta = 1.0f, float alpha = 1.0f);
+
+void addbmm_(Tensor output, Tensor input, Tensor batch1, Tensor batch2, float beta, float alpha);
+
+} // namespace infinicore::op

include/infinicore/ops/affine_grid.hpp

@@ -0,0 +1,17 @@
+#pragma once
+
+#include "../device.hpp"
+#include "common/op.hpp"
+#include <vector>
+
+namespace infinicore::op {
+
+class AffineGrid {
+public:
+    using schema = void (*)(Tensor, Tensor, bool);
+    static void execute(Tensor output, Tensor theta, bool align_corners);
+    static common::OpDispatcher<schema> &dispatcher();
+};
+Tensor affine_grid(Tensor theta, const std::vector<int64_t>& size, bool align_corners = false);
+
+} // namespace infinicore::op

include/infinicore/ops/floor.hpp

@@ -0,0 +1,16 @@
+#pragma once
+
+#include "../device.hpp"
+#include "common/op.hpp"
+
+namespace infinicore::op {
+class Floor {
+public:
+    using schema = void (*)(Tensor, Tensor);
+    static void execute(Tensor output, Tensor input);
+    static common::OpDispatcher<schema> &dispatcher();
+};
+
+Tensor floor(Tensor input);
+void floor_(Tensor output, Tensor input);
+} // namespace infinicore::op

include/infiniop.h

@@ -12,6 +12,12 @@
 #include "infiniop/ops/asin.h"
 #include "infiniop/ops/asinh.h"
 #include "infiniop/ops/atanh.h"
+#include "infiniop/ops/acos.h"
+#include "infiniop/ops/add.h"
+#include "infiniop/ops/adaptive_avg_pool1d.h"
+#include "infiniop/ops/addbmm.h"
+#include "infiniop/ops/affine_grid.h"
+#include "infiniop/ops/floor.h"
 #include "infiniop/ops/attention.h"
 #include "infiniop/ops/avg_pool1d.h"
 #include "infiniop/ops/binary_cross_entropy_with_logits.h"

include/infiniop/ops/acos.h

@@ -0,0 +1,24 @@
+#ifndef __INFINIOP_ACOS_API_H__
+#define __INFINIOP_ACOS_API_H__
+
+#include "../operator_descriptor.h"
+
+typedef struct InfiniopDescriptor *infiniopAcosDescriptor_t;
+
+__C __export infiniStatus_t infiniopCreateAcosDescriptor(infiniopHandle_t handle,
+                                                         infiniopAcosDescriptor_t *desc_ptr,
+                                                         infiniopTensorDescriptor_t y, 
+                                                         infiniopTensorDescriptor_t x );
+
+__C __export infiniStatus_t infiniopGetAcosWorkspaceSize(infiniopAcosDescriptor_t desc, size_t *size);
+
+__C __export infiniStatus_t infiniopAcos(infiniopAcosDescriptor_t desc,
+                                         void *workspace,
+                                         size_t workspace_size,
+                                         void *y,      
+                                         const void *x, 
+                                         void *stream);
+
+__C __export infiniStatus_t infiniopDestroyAcosDescriptor(infiniopAcosDescriptor_t desc);
+
+#endif 

include/infiniop/ops/adaptive_avg_pool1d.h

@@ -0,0 +1,34 @@
+#ifndef __INFINIOP_ADAPTIVE_AVG_POOL1D_API_H__
+#define __INFINIOP_ADAPTIVE_AVG_POOL1D_API_H__
+
+#include "../operator_descriptor.h"
+
+// 定义算子描述符类型
+typedef struct InfiniopDescriptor *infiniopAdaptiveAvgPool1dDescriptor_t;
+
+// 1. 创建算子描述符
+__C __export infiniStatus_t infiniopCreateAdaptiveAvgPool1dDescriptor(
+    infiniopHandle_t handle,
+    infiniopAdaptiveAvgPool1dDescriptor_t *desc_ptr,
+    infiniopTensorDescriptor_t out_desc,
+    infiniopTensorDescriptor_t input_desc);
+
+// 2. 获取 Workspace 大小
+__C __export infiniStatus_t infiniopGetAdaptiveAvgPool1dWorkspaceSize(
+    infiniopAdaptiveAvgPool1dDescriptor_t desc, 
+    size_t *size);
+
+// 3. 执行计算
+__C __export infiniStatus_t infiniopAdaptiveAvgPool1d(
+    infiniopAdaptiveAvgPool1dDescriptor_t desc,
+    void *workspace,
+    size_t workspace_size,
+    void *output,
+    const void *input,
+    void *stream);
+
+// 4. 销毁描述符
+__C __export infiniStatus_t infiniopDestroyAdaptiveAvgPool1dDescriptor(
+    infiniopAdaptiveAvgPool1dDescriptor_t desc);
+
+#endif // __INFINIOP_ADAPTIVE_AVG_POOL1D_API_H__

include/infiniop/ops/addbmm.h

@@ -0,0 +1,30 @@
+#ifndef __INFINIOP_ADDBMM_API_H__
+#define __INFINIOP_ADDBMM_API_H__
+
+#include "../operator_descriptor.h"
+
+typedef struct InfiniopDescriptor *infiniopAddbmmDescriptor_t;
+
+__C __export infiniStatus_t infiniopCreateAddbmmDescriptor(infiniopHandle_t handle,
+                                                           infiniopAddbmmDescriptor_t *desc_ptr,
+                                                           infiniopTensorDescriptor_t out_desc,
+                                                           infiniopTensorDescriptor_t input_desc,
+                                                           infiniopTensorDescriptor_t batch1_desc,
+                                                           infiniopTensorDescriptor_t batch2_desc,
+                                                           float alpha,
+                                                           float beta);
+
+__C __export infiniStatus_t infiniopGetAddbmmWorkspaceSize(infiniopAddbmmDescriptor_t desc, size_t *size);
+
+__C __export infiniStatus_t infiniopAddbmm(infiniopAddbmmDescriptor_t desc,
+                                           void *workspace,
+                                           size_t workspace_size,
+                                           void *output,
+                                           const void *input,
+                                           const void *batch1,
+                                           const void *batch2,
+                                           void *stream);
+
+__C __export infiniStatus_t infiniopDestroyAddbmmDescriptor(infiniopAddbmmDescriptor_t desc);
+
+#endif // __INFINIOP_ADDBMM_API_H__

include/infiniop/ops/affine_grid.h

@@ -0,0 +1,25 @@
+#ifndef __INFINIOP_AFFINE_GRID_API_H__
+#define __INFINIOP_AFFINE_GRID_API_H__
+#include <stdint.h>
+#include "../operator_descriptor.h"
+
+typedef struct InfiniopDescriptor *infiniopAffineGridDescriptor_t;
+
+__C __export infiniStatus_t infiniopCreateAffineGridDescriptor(infiniopHandle_t handle,
+                                                               infiniopAffineGridDescriptor_t *desc_ptr,
+                                                               infiniopTensorDescriptor_t output_desc,
+                                                               infiniopTensorDescriptor_t input_desc,
+                                                               uint8_t align_corners);
+
+__C __export infiniStatus_t infiniopGetAffineGridWorkspaceSize(infiniopAffineGridDescriptor_t desc, size_t *size);
+
+__C __export infiniStatus_t infiniopAffineGrid(infiniopAffineGridDescriptor_t desc,
+                                               void *workspace,
+                                               size_t workspace_size,
+                                               void *output,
+                                               const void *input,
+                                               void *stream);
+
+__C __export infiniStatus_t infiniopDestroyAffineGridDescriptor(infiniopAffineGridDescriptor_t desc);
+
+#endif // __INFINIOP_AFFINE_GRID_API_H__