issue/1031 merge T1-1-37

Author: LaiQuan-conquer

include/infinicore/ops/diff.hpp

@@ -0,0 +1,15 @@
+#pragma once
+
+#include "../device.hpp"
+#include "../graph/graph.hpp"
+#include "common/op.hpp"
+
+namespace infinicore::op {
+
+INFINICORE_GRAPH_OP_CLASS(Diff, Tensor, const Tensor &, int, int);
+
+Tensor diff(const Tensor &x, int n = 1, int dim = -1);
+void diff_(Tensor y, const Tensor &x, int n = 1, int dim = -1);
+
+} // namespace infinicore::op
+

include/infinicore/ops/digamma.hpp

@@ -0,0 +1,15 @@
+#pragma once
+
+#include "../device.hpp"
+#include "../graph/graph.hpp"
+#include "common/op.hpp"
+
+namespace infinicore::op {
+
+INFINICORE_GRAPH_OP_CLASS(Digamma, Tensor, const Tensor &);
+
+Tensor digamma(const Tensor &x);
+void digamma_(Tensor y, const Tensor &x);
+
+} // namespace infinicore::op
+

include/infinicore/ops/dist.hpp

@@ -0,0 +1,15 @@
+#pragma once
+
+#include "../device.hpp"
+#include "../graph/graph.hpp"
+#include "common/op.hpp"
+
+namespace infinicore::op {
+
+INFINICORE_GRAPH_OP_CLASS(Dist, Tensor, const Tensor &, const Tensor &, double);
+
+Tensor dist(const Tensor &x1, const Tensor &x2, double p = 2.0);
+void dist_(Tensor y, const Tensor &x1, const Tensor &x2, double p = 2.0);
+
+} // namespace infinicore::op
+

include/infinicore/ops/logdet.hpp

@@ -0,0 +1,15 @@
+#pragma once
+
+#include "../device.hpp"
+#include "../graph/graph.hpp"
+#include "common/op.hpp"
+
+namespace infinicore::op {
+
+INFINICORE_GRAPH_OP_CLASS(Logdet, Tensor, const Tensor &);
+
+Tensor logdet(const Tensor &x);
+void logdet_(Tensor y, const Tensor &x);
+
+} // namespace infinicore::op
+

include/infinicore/ops/pad.hpp

@@ -0,0 +1,26 @@
+#pragma once
+
+#include "../device.hpp"
+#include "../graph/graph.hpp"
+#include "common/op.hpp"
+
+#include <string>
+#include <vector>
+
+namespace infinicore::op {
+
+INFINICORE_GRAPH_OP_CLASS(Pad, Tensor, const Tensor &, const std::vector<int> &, const std::string &, double);
+
+Tensor pad(const Tensor &x,
+           const std::vector<int> &pad,
+           const std::string &mode = "constant",
+           double value = 0.0);
+
+void pad_(Tensor y,
+          const Tensor &x,
+          const std::vector<int> &pad,
+          const std::string &mode = "constant",
+          double value = 0.0);
+
+} // namespace infinicore::op
+

include/infiniop.h

@@ -29,6 +29,9 @@
 #include "infiniop/ops/dequant/per_tensor_dequant_int8.h"
 #include "infiniop/ops/dequantize_awq.h"
 #include "infiniop/ops/dequantize_gptq.h"
+#include "infiniop/ops/diff.h"
+#include "infiniop/ops/digamma.h"
+#include "infiniop/ops/dist.h"
 #include "infiniop/ops/embedding.h"
 #include "infiniop/ops/equal.h"
 #include "infiniop/ops/flash_attention.h"
@@ -59,11 +62,13 @@
 #include "infiniop/ops/logaddexp.h"
 #include "infiniop/ops/logaddexp2.h"
 #include "infiniop/ops/logcumsumexp.h"
+#include "infiniop/ops/logdet.h"
 #include "infiniop/ops/lp_norm.h"
 #include "infiniop/ops/masked_select.h"
 #include "infiniop/ops/mul.h"
 #include "infiniop/ops/multi_margin_loss.h"
 #include "infiniop/ops/ones.h"
+#include "infiniop/ops/pad.h"
 #include "infiniop/ops/paged_attention.h"
 #include "infiniop/ops/paged_attention_prefill.h"
 #include "infiniop/ops/paged_caching.h"

include/infiniop/ops/diff.h

@@ -0,0 +1,26 @@
+#ifndef __INFINIOP_DIFF_API_H__
+#define __INFINIOP_DIFF_API_H__
+
+#include "../operator_descriptor.h"
+
+typedef struct InfiniopDescriptor *infiniopDiffDescriptor_t;
+
+__C __export infiniStatus_t infiniopCreateDiffDescriptor(infiniopHandle_t handle,
+                                                         infiniopDiffDescriptor_t *desc_ptr,
+                                                         infiniopTensorDescriptor_t y,
+                                                         infiniopTensorDescriptor_t x,
+                                                         int dim,
+                                                         int n);
+
+__C __export infiniStatus_t infiniopGetDiffWorkspaceSize(infiniopDiffDescriptor_t desc, size_t *size);
+
+__C __export infiniStatus_t infiniopDiff(infiniopDiffDescriptor_t desc,
+                                         void *workspace,
+                                         size_t workspace_size,
+                                         void *y,
+                                         const void *x,
+                                         void *stream);
+
+__C __export infiniStatus_t infiniopDestroyDiffDescriptor(infiniopDiffDescriptor_t desc);
+
+#endif

include/infiniop/ops/digamma.h

@@ -0,0 +1,24 @@
+#ifndef __INFINIOP_DIGAMMA_API_H__
+#define __INFINIOP_DIGAMMA_API_H__
+
+#include "../operator_descriptor.h"
+
+typedef struct InfiniopDescriptor *infiniopDigammaDescriptor_t;
+
+__C __export infiniStatus_t infiniopCreateDigammaDescriptor(infiniopHandle_t handle,
+                                                            infiniopDigammaDescriptor_t *desc_ptr,
+                                                            infiniopTensorDescriptor_t y,
+                                                            infiniopTensorDescriptor_t x);
+
+__C __export infiniStatus_t infiniopGetDigammaWorkspaceSize(infiniopDigammaDescriptor_t desc, size_t *size);
+
+__C __export infiniStatus_t infiniopDigamma(infiniopDigammaDescriptor_t desc,
+                                            void *workspace,
+                                            size_t workspace_size,
+                                            void *y,
+                                            const void *x,
+                                            void *stream);
+
+__C __export infiniStatus_t infiniopDestroyDigammaDescriptor(infiniopDigammaDescriptor_t desc);
+
+#endif

include/infiniop/ops/dist.h

@@ -0,0 +1,27 @@
+#ifndef __INFINIOP_DIST_API_H__
+#define __INFINIOP_DIST_API_H__
+
+#include "../operator_descriptor.h"
+
+typedef struct InfiniopDescriptor *infiniopDistDescriptor_t;
+
+__C __export infiniStatus_t infiniopCreateDistDescriptor(infiniopHandle_t handle,
+                                                         infiniopDistDescriptor_t *desc_ptr,
+                                                         infiniopTensorDescriptor_t y,
+                                                         infiniopTensorDescriptor_t x1,
+                                                         infiniopTensorDescriptor_t x2,
+                                                         double p);
+
+__C __export infiniStatus_t infiniopGetDistWorkspaceSize(infiniopDistDescriptor_t desc, size_t *size);
+
+__C __export infiniStatus_t infiniopDist(infiniopDistDescriptor_t desc,
+                                         void *workspace,
+                                         size_t workspace_size,
+                                         void *y,
+                                         const void *x1,
+                                         const void *x2,
+                                         void *stream);
+
+__C __export infiniStatus_t infiniopDestroyDistDescriptor(infiniopDistDescriptor_t desc);
+
+#endif

include/infiniop/ops/logdet.h

@@ -0,0 +1,24 @@
+#ifndef __INFINIOP_LOGDET_API_H__
+#define __INFINIOP_LOGDET_API_H__
+
+#include "../operator_descriptor.h"
+
+typedef struct InfiniopDescriptor *infiniopLogdetDescriptor_t;
+
+__C __export infiniStatus_t infiniopCreateLogdetDescriptor(infiniopHandle_t handle,
+                                                           infiniopLogdetDescriptor_t *desc_ptr,
+                                                           infiniopTensorDescriptor_t y,
+                                                           infiniopTensorDescriptor_t x);
+
+__C __export infiniStatus_t infiniopGetLogdetWorkspaceSize(infiniopLogdetDescriptor_t desc, size_t *size);
+
+__C __export infiniStatus_t infiniopLogdet(infiniopLogdetDescriptor_t desc,
+                                           void *workspace,
+                                           size_t workspace_size,
+                                           void *y,
+                                           const void *x,
+                                           void *stream);
+
+__C __export infiniStatus_t infiniopDestroyLogdetDescriptor(infiniopLogdetDescriptor_t desc);
+
+#endif