Skip to content

Commit 02dc83e

Browse files
PanZezhong1725PanZezhong1725
committed
issue/1031 fix T1-1-2
1 parent b37636c commit 02dc83e

Some content is hidden

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

50 files changed

+390
-471
lines changed

‎include/infiniop/ops/avg_pool3d.h‎

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -5,23 +5,23 @@
55

66
typedef struct InfiniopDescriptor *infiniopAvgPool3dDescriptor_t;
77

8-
__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);
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);
1515

16-
__C __export infiniStatus_t infiniopGetAvgPool3dWorkspaceSize(infiniopAvgPool3dDescriptor_t desc, size_t *size);
16+
__INFINI_C __export infiniStatus_t infiniopGetAvgPool3dWorkspaceSize(infiniopAvgPool3dDescriptor_t desc, size_t *size);
1717

18-
__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);
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);
2424

25-
__C __export infiniStatus_t infiniopDestroyAvgPool3dDescriptor(infiniopAvgPool3dDescriptor_t desc);
25+
__INFINI_C __export infiniStatus_t infiniopDestroyAvgPool3dDescriptor(infiniopAvgPool3dDescriptor_t desc);
2626

2727
#endif

‎include/infiniop/ops/dot.h‎

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -5,22 +5,22 @@
55

66
typedef struct InfiniopDescriptor *infiniopDotDescriptor_t;
77

8-
__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);
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);
1313

14-
__C __export infiniStatus_t infiniopGetDotWorkspaceSize(infiniopDotDescriptor_t desc, size_t *size);
14+
__INFINI_C __export infiniStatus_t infiniopGetDotWorkspaceSize(infiniopDotDescriptor_t desc, size_t *size);
1515

16-
__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);
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);
2323

24-
__C __export infiniStatus_t infiniopDestroyDotDescriptor(infiniopDotDescriptor_t desc);
24+
__INFINI_C __export infiniStatus_t infiniopDestroyDotDescriptor(infiniopDotDescriptor_t desc);
2525

2626
#endif

‎include/infiniop/ops/histc.h‎

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -6,23 +6,23 @@
66

77
typedef struct InfiniopDescriptor *infiniopHistcDescriptor_t;
88

9-
__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);
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);
1616

17-
__C __export infiniStatus_t infiniopGetHistcWorkspaceSize(infiniopHistcDescriptor_t desc, size_t *size);
17+
__INFINI_C __export infiniStatus_t infiniopGetHistcWorkspaceSize(infiniopHistcDescriptor_t desc, size_t *size);
1818

19-
__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);
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);
2525

26-
__C __export infiniStatus_t infiniopDestroyHistcDescriptor(infiniopHistcDescriptor_t desc);
26+
__INFINI_C __export infiniStatus_t infiniopDestroyHistcDescriptor(infiniopHistcDescriptor_t desc);
2727

2828
#endif

‎include/infiniop/ops/log10.h‎

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -5,20 +5,20 @@
55

66
typedef struct InfiniopDescriptor *infiniopLog10Descriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateLog10Descriptor(infiniopHandle_t handle,
9-
infiniopLog10Descriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t y,
11-
infiniopTensorDescriptor_t x);
8+
__INFINI_C __export infiniStatus_t infiniopCreateLog10Descriptor(infiniopHandle_t handle,
9+
infiniopLog10Descriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t y,
11+
infiniopTensorDescriptor_t x);
1212

13-
__C __export infiniStatus_t infiniopGetLog10WorkspaceSize(infiniopLog10Descriptor_t desc, size_t *size);
13+
__INFINI_C __export infiniStatus_t infiniopGetLog10WorkspaceSize(infiniopLog10Descriptor_t desc, size_t *size);
1414

15-
__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);
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);
2121

22-
__C __export infiniStatus_t infiniopDestroyLog10Descriptor(infiniopLog10Descriptor_t desc);
22+
__INFINI_C __export infiniStatus_t infiniopDestroyLog10Descriptor(infiniopLog10Descriptor_t desc);
2323

2424
#endif

‎include/infiniop/ops/log1p.h‎

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -5,20 +5,20 @@
55

66
typedef struct InfiniopDescriptor *infiniopLog1pDescriptor_t;
77

8-
__C __export infiniStatus_t infiniopCreateLog1pDescriptor(infiniopHandle_t handle,
9-
infiniopLog1pDescriptor_t *desc_ptr,
10-
infiniopTensorDescriptor_t y,
11-
infiniopTensorDescriptor_t x);
8+
__INFINI_C __export infiniStatus_t infiniopCreateLog1pDescriptor(infiniopHandle_t handle,
9+
infiniopLog1pDescriptor_t *desc_ptr,
10+
infiniopTensorDescriptor_t y,
11+
infiniopTensorDescriptor_t x);
1212

13-
__C __export infiniStatus_t infiniopGetLog1pWorkspaceSize(infiniopLog1pDescriptor_t desc, size_t *size);
13+
__INFINI_C __export infiniStatus_t infiniopGetLog1pWorkspaceSize(infiniopLog1pDescriptor_t desc, size_t *size);
1414

15-
__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);
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);
2121

22-
__C __export infiniStatus_t infiniopDestroyLog1pDescriptor(infiniopLog1pDescriptor_t desc);
22+
__INFINI_C __export infiniStatus_t infiniopDestroyLog1pDescriptor(infiniopLog1pDescriptor_t desc);
2323

2424
#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

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

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
#ifndef __INFINIOP_MOORE_KERNEL_COMMON_H__
2+
#define __INFINIOP_MOORE_KERNEL_COMMON_H__
13
#define INFINIOP_MOORE_KERNEL __global__ void
24

35
#include <musa_bf16.h>
@@ -72,3 +74,4 @@ exp_(const __mt_bfloat16 x) {
7274
float f_result = expf(f_val);
7375
return __float2bfloat16(f_result);
7476
}
77+
#endif

0 commit comments

Comments
 (0)