Skip to content

Commit d6ab557

Browse files
PanZezhong1725PanZezhong1725
authored
Merge pull request #1115 from InfiniTensor/issue/1031_T1-1-2
【比赛2025秋】T1-1-2
2 parents 6c72ea3 + 02dc83e commit d6ab557

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

71 files changed

+4879
-5
lines changed

‎include/infiniop.h‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
#include "infiniop/ops/atanh.h"
1919
#include "infiniop/ops/attention.h"
2020
#include "infiniop/ops/avg_pool1d.h"
21+
#include "infiniop/ops/avg_pool3d.h"
2122
#include "infiniop/ops/binary_cross_entropy_with_logits.h"
2223
#include "infiniop/ops/block_diag.h"
2324
#include "infiniop/ops/broadcast_to.h"
@@ -32,6 +33,7 @@
3233
#include "infiniop/ops/diff.h"
3334
#include "infiniop/ops/digamma.h"
3435
#include "infiniop/ops/dist.h"
36+
#include "infiniop/ops/dot.h"
3537
#include "infiniop/ops/embedding.h"
3638
#include "infiniop/ops/equal.h"
3739
#include "infiniop/ops/flash_attention.h"
@@ -46,6 +48,7 @@
4648
#include "infiniop/ops/hardswish.h"
4749
#include "infiniop/ops/hardtanh.h"
4850
#include "infiniop/ops/hinge_embedding_loss.h"
51+
#include "infiniop/ops/histc.h"
4952
#include "infiniop/ops/huber_loss.h"
5053
#include "infiniop/ops/hypot.h"
5154
#include "infiniop/ops/index_add.h"
@@ -58,6 +61,8 @@
5861
#include "infiniop/ops/layer_norm.h"
5962
#include "infiniop/ops/ldexp.h"
6063
#include "infiniop/ops/lerp.h"
64+
#include "infiniop/ops/log10.h"
65+
#include "infiniop/ops/log1p.h"
6166
#include "infiniop/ops/log_softmax.h"
6267
#include "infiniop/ops/logaddexp.h"
6368
#include "infiniop/ops/logaddexp2.h"

‎include/infiniop/ops/avg_pool3d.h‎

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,27 @@
1+
#ifndef __INFINIOP_AVG_POOL3D_API_H__
2+
#define __INFINIOP_AVG_POOL3D_API_H__
3+
4+
#include "../operator_descriptor.h"
5+
6+
typedef struct InfiniopDescriptor *infiniopAvgPool3dDescriptor_t;
7+
8+
__INFINI_C __export infiniStatus_t infiniopCreateAvgPool3dDescriptor(infiniopHandle_t handle,
9+
infiniopAvgPool3dDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t y,
11+
infiniopTensorDescriptor_t x,
12+
void *kernel_size,
13+
void *stride,
14+
void *padding);
15+
16+
__INFINI_C __export infiniStatus_t infiniopGetAvgPool3dWorkspaceSize(infiniopAvgPool3dDescriptor_t desc, size_t *size);
17+
18+
__INFINI_C __export infiniStatus_t infiniopAvgPool3d(infiniopAvgPool3dDescriptor_t desc,
19+
void *workspace,
20+
size_t workspace_size,
21+
void *y,
22+
const void *x,
23+
void *stream);
24+
25+
__INFINI_C __export infiniStatus_t infiniopDestroyAvgPool3dDescriptor(infiniopAvgPool3dDescriptor_t desc);
26+
27+
#endif

‎include/infiniop/ops/dot.h‎

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,26 @@
1+
#ifndef __INFINIOP_DOT_API_H__
2+
#define __INFINIOP_DOT_API_H__
3+
4+
#include "../operator_descriptor.h"
5+
6+
typedef struct InfiniopDescriptor *infiniopDotDescriptor_t;
7+
8+
__INFINI_C __export infiniStatus_t infiniopCreateDotDescriptor(infiniopHandle_t handle,
9+
infiniopDotDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t y,
11+
infiniopTensorDescriptor_t a,
12+
infiniopTensorDescriptor_t b);
13+
14+
__INFINI_C __export infiniStatus_t infiniopGetDotWorkspaceSize(infiniopDotDescriptor_t desc, size_t *size);
15+
16+
__INFINI_C __export infiniStatus_t infiniopDot(infiniopDotDescriptor_t desc,
17+
void *workspace,
18+
size_t workspace_size,
19+
void *y,
20+
const void *a,
21+
const void *b,
22+
void *stream);
23+
24+
__INFINI_C __export infiniStatus_t infiniopDestroyDotDescriptor(infiniopDotDescriptor_t desc);
25+
26+
#endif

‎include/infiniop/ops/histc.h‎

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
#ifndef __INFINIOP_HISTC_API_H__
2+
#define __INFINIOP_HISTC_API_H__
3+
4+
#include "../operator_descriptor.h"
5+
#include <stdint.h>
6+
7+
typedef struct InfiniopDescriptor *infiniopHistcDescriptor_t;
8+
9+
__INFINI_C __export infiniStatus_t infiniopCreateHistcDescriptor(infiniopHandle_t handle,
10+
infiniopHistcDescriptor_t *desc_ptr,
11+
infiniopTensorDescriptor_t y,
12+
infiniopTensorDescriptor_t x,
13+
int64_t bins,
14+
double min_val,
15+
double max_val);
16+
17+
__INFINI_C __export infiniStatus_t infiniopGetHistcWorkspaceSize(infiniopHistcDescriptor_t desc, size_t *size);
18+
19+
__INFINI_C __export infiniStatus_t infiniopHistc(infiniopHistcDescriptor_t desc,
20+
void *workspace,
21+
size_t workspace_size,
22+
void *y,
23+
const void *x,
24+
void *stream);
25+
26+
__INFINI_C __export infiniStatus_t infiniopDestroyHistcDescriptor(infiniopHistcDescriptor_t desc);
27+
28+
#endif

‎include/infiniop/ops/log10.h‎

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
#ifndef __INFINIOP_LOG10_API_H__
2+
#define __INFINIOP_LOG10_API_H__
3+
4+
#include "../operator_descriptor.h"
5+
6+
typedef struct InfiniopDescriptor *infiniopLog10Descriptor_t;
7+
8+
__INFINI_C __export infiniStatus_t infiniopCreateLog10Descriptor(infiniopHandle_t handle,
9+
infiniopLog10Descriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t y,
11+
infiniopTensorDescriptor_t x);
12+
13+
__INFINI_C __export infiniStatus_t infiniopGetLog10WorkspaceSize(infiniopLog10Descriptor_t desc, size_t *size);
14+
15+
__INFINI_C __export infiniStatus_t infiniopLog10(infiniopLog10Descriptor_t desc,
16+
void *workspace,
17+
size_t workspace_size,
18+
void *y,
19+
const void *x,
20+
void *stream);
21+
22+
__INFINI_C __export infiniStatus_t infiniopDestroyLog10Descriptor(infiniopLog10Descriptor_t desc);
23+
24+
#endif

‎include/infiniop/ops/log1p.h‎

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
#ifndef __INFINIOP_LOG1P_API_H__
2+
#define __INFINIOP_LOG1P_API_H__
3+
4+
#include "../operator_descriptor.h"
5+
6+
typedef struct InfiniopDescriptor *infiniopLog1pDescriptor_t;
7+
8+
__INFINI_C __export infiniStatus_t infiniopCreateLog1pDescriptor(infiniopHandle_t handle,
9+
infiniopLog1pDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t y,
11+
infiniopTensorDescriptor_t x);
12+
13+
__INFINI_C __export infiniStatus_t infiniopGetLog1pWorkspaceSize(infiniopLog1pDescriptor_t desc, size_t *size);
14+
15+
__INFINI_C __export infiniStatus_t infiniopLog1p(infiniopLog1pDescriptor_t desc,
16+
void *workspace,
17+
size_t workspace_size,
18+
void *y,
19+
const void *x,
20+
void *stream);
21+
22+
__INFINI_C __export infiniStatus_t infiniopDestroyLog1pDescriptor(infiniopLog1pDescriptor_t desc);
23+
24+
#endif

‎src/infiniop/devices/metax/metax_common.h‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
#pragma once
12
#include "../../../utils.h"
23
#include "../pool.h"
34
#include "metax_handle.h"

‎src/infiniop/devices/metax/metax_ht2mc.h‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,8 +61,10 @@
6161
#define hcdnnTensorDescriptor_t mcdnnTensorDescriptor_t
6262
#define hcdnnStatus_t mcdnnStatus_t
6363
#define hcdnnSetTensor4dDescriptor mcdnnSetTensor4dDescriptor
64+
#define hcdnnSetTensorNdDescriptor mcdnnSetTensorNdDescriptor
6465
#define hcdnnSetStream mcdnnSetStream
6566
#define hcdnnSetPooling2dDescriptor mcdnnSetPooling2dDescriptor
67+
#define hcdnnSetPoolingNdDescriptor mcdnnSetPoolingNdDescriptor
6668
#define hcdnnPoolingMode_t mcdnnPoolingMode_t
6769
#define hcdnnPoolingForward mcdnnPoolingForward
6870
#define hcdnnPoolingDescriptor_t mcdnnPoolingDescriptor_t

‎src/infiniop/devices/metax/metax_kernel_common.h‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
#pragma once
12
#define INFINIOP_METAX_KERNEL __global__ void
23

34
#ifdef ENABLE_METAX_MC_API

‎src/infiniop/devices/moore/moore_common.h‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
#ifndef __INFINIOP_MOORE_COMMON_H__
2+
#define __INFINIOP_MOORE_COMMON_H__
13
#include "../../../utils.h"
24
#include "../pool.h"
35
#include "moore_handle.h"
@@ -42,3 +44,4 @@ class Handle::Internal {
4244
};
4345

4446
} // namespace device::moore
47+
#endif

0 commit comments

Comments
 (0)