"setup.py" did not exist on "1f5074062b638bcc107c09d5086d62327c07608d"
matmul.h 1.35 KB
Newer Older
PanZezhong's avatar
PanZezhong committed
1
2
#ifndef __INFINIOP_MATMUL_API_H__
#define __INFINIOP_MATMUL_API_H__
PanZezhongQY's avatar
PanZezhongQY committed
3

PanZezhong's avatar
PanZezhong committed
4
#include "../operator_descriptor.h"
PanZezhongQY's avatar
PanZezhongQY committed
5
6
7

typedef InfiniopDescriptor *infiniopMatmulDescriptor_t;

PanZezhong's avatar
PanZezhong committed
8
9
10
11
12
__C __export infiniStatus_t infiniopCreateMatmulDescriptor(infiniopHandle_t handle,
                                                           infiniopMatmulDescriptor_t *desc_ptr,
                                                           infiniopTensorDescriptor_t c_desc,
                                                           infiniopTensorDescriptor_t a_desc,
                                                           infiniopTensorDescriptor_t b_desc);
PanZezhongQY's avatar
PanZezhongQY committed
13

PanZezhong's avatar
PanZezhong committed
14
__C __export infiniStatus_t infiniopGetMatmulWorkspaceSize(infiniopMatmulDescriptor_t desc, size_t *size);
PanZezhongQY's avatar
PanZezhongQY committed
15

PanZezhong's avatar
PanZezhong committed
16
17
18
19
20
21
22
23
24
__C __export infiniStatus_t infiniopMatmul(infiniopMatmulDescriptor_t desc,
                                           void *workspace,
                                           size_t workspace_size,
                                           void *c,
                                           void const *a,
                                           void const *b,
                                           float alpha,
                                           float beta,
                                           void *stream);
PanZezhongQY's avatar
PanZezhongQY committed
25

PanZezhong's avatar
PanZezhong committed
26
__C __export infiniStatus_t infiniopDestroyMatmulDescriptor(infiniopMatmulDescriptor_t desc);
PanZezhongQY's avatar
PanZezhongQY committed
27
28

#endif