Skip to content

Commit 32bd65d

Browse files
[SYCLomatic][cuTENSOR] Added initial inc file for cuTensor APIs (#2754)
1 parent 16d76bc commit 32bd65d

File tree

1 file changed

+124
-0
lines changed

1 file changed

+124
-0
lines changed
Lines changed: 124 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,124 @@
1+
//===------------------------ APINames_cuTENSOR.inc -----------------------===//
2+
//
3+
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4+
// See https://llvm.org/LICENSE.txt for license information.
5+
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6+
//
7+
//===----------------------------------------------------------------------===//
8+
// clang-format off
9+
10+
/****************************************************************************
11+
*
12+
* ENTRY(Interface APIName, APINAME, ProcessedOrNot, Flag, Priority, MigrationDesc)
13+
* Interface APIName: name of CUDA Interface API available in official document
14+
* APIName: name of CUDA API
15+
*
16+
* ProcessedOrNot: false|true
17+
* false: this API has not been processed.
18+
* true: this API has been processed, need check MigrationDesc field to
19+
see if it has be migrated Commently.
20+
*
21+
* Flag: Bit map for all kind of flag for this API.
22+
* default(all bits are 0): NO_FLAG
23+
* bit 0: API_CALL_REMOVED
24+
* bit 1: API_CALL_UNSUPPORTED
25+
* Priority: migration priority, eg. P0 - P4
26+
* P0: should support in Beta release.
27+
* P4: should support in Gold.
28+
*
29+
* MigrationDesc: Comment[: Comment] | Unsupported: Comment
30+
* Comment: this API has been migrated.
31+
* Unsupported: this API can not be migrated currently.
32+
* Must add Comment to explain why this API is Unsupported.
33+
* Comment: extra comment on the migration.
34+
*
35+
****************************************************************************/
36+
37+
// Helper Functions
38+
ENTRY(cutensorCreate, cutensorCreate, false, NO_FLAG, P4, "comment")
39+
ENTRY(cutensorDestroy, cutensorDestroy, false, NO_FLAG, P4, "comment")
40+
ENTRY(cutensorCreateTensorDescriptor, cutensorCreateTensorDescriptor, false, NO_FLAG, P4, "comment")
41+
ENTRY(cutensorDestroyTensorDescriptor, cutensorDestroyTensorDescriptor, false, NO_FLAG, P4, "comment")
42+
ENTRY(cutensorGetErrorString, cutensorGetErrorString, false, NO_FLAG, P4, "comment")
43+
ENTRY(cutensorGetVersion, cutensorGetVersion, false, NO_FLAG, P4, "comment")
44+
ENTRY(cutensorGetCudartVersion, cutensorGetCudartVersion, false, NO_FLAG, P4, "comment")
45+
46+
47+
// Element-wise Operations
48+
ENTRY(cutensorCreateElementwiseTrinary, cutensorCreateElementwiseTrinary, false, NO_FLAG, P4, "comment")
49+
ENTRY(cutensorElementwiseTrinaryExecute, cutensorElementwiseTrinaryExecute, false, NO_FLAG, P4, "comment")
50+
ENTRY(cutensorCreateElementwiseBinary, cutensorCreateElementwiseBinary, false, NO_FLAG, P4, "comment")
51+
ENTRY(cutensorElementwiseBinaryExecute, cutensorElementwiseBinaryExecute, false, NO_FLAG, P4, "comment")
52+
ENTRY(cutensorCreatePermutation, cutensorCreatePermutation, false, NO_FLAG, P4, "comment")
53+
ENTRY(cutensorPermute, cutensorPermute, false, NO_FLAG, P4, "comment")
54+
55+
56+
// Contraction Operations
57+
ENTRY(cutensorCreateContraction, cutensorCreateContraction, false, NO_FLAG, P4, "comment")
58+
ENTRY(cutensorContract, cutensorContract, false, NO_FLAG, P4, "comment")
59+
ENTRY(cutensorCreateContractionTrinary, cutensorCreateContractionTrinary, false, NO_FLAG, P4, "comment")
60+
ENTRY(cutensorContractTrinary, cutensorContractTrinary, false, NO_FLAG, P4, "comment")
61+
62+
63+
// Reduction Operations
64+
ENTRY(cutensorCreateReduction, cutensorCreateReduction, false, NO_FLAG, P4, "comment")
65+
ENTRY(cutensorReduce, cutensorReduce, false, NO_FLAG, P4, "comment")
66+
67+
68+
// Generic Operation Functions
69+
ENTRY(cutensorDestroyOperationDescriptor, cutensorDestroyOperationDescriptor, false, NO_FLAG, P4, "comment")
70+
ENTRY(cutensorOperationDescriptorGetAttribute, cutensorOperationDescriptorGetAttribute, false, NO_FLAG, P4, "comment")
71+
ENTRY(cutensorOperationDescriptorSetAttribute, cutensorOperationDescriptorSetAttribute, false, NO_FLAG, P4, "comment")
72+
ENTRY(cutensorCreatePlanPreference, cutensorCreatePlanPreference, false, NO_FLAG, P4, "comment")
73+
ENTRY(cutensorDestroyPlanPreference, cutensorDestroyPlanPreference, false, NO_FLAG, P4, "comment")
74+
ENTRY(cutensorPlanPreferenceSetAttribute, cutensorPlanPreferenceSetAttribute, false, NO_FLAG, P4, "comment")
75+
ENTRY(cutensorEstimateWorkspaceSize, cutensorEstimateWorkspaceSize, false, NO_FLAG, P4, "comment")
76+
ENTRY(cutensorCreatePlan, cutensorCreatePlan, false, NO_FLAG, P4, "comment")
77+
ENTRY(cutensorDestroyPlan, cutensorDestroyPlan, false, NO_FLAG, P4, "comment")
78+
ENTRY(cutensorPlanGetAttribute, cutensorPlanGetAttribute, false, NO_FLAG, P4, "comment")
79+
80+
81+
// Cache-related Operations
82+
ENTRY(cutensorHandleResizePlanCache, cutensorHandleResizePlanCache, false, NO_FLAG, P4, "comment")
83+
ENTRY(cutensorHandleReadPlanCacheFromFile, cutensorHandleReadPlanCacheFromFile, false, NO_FLAG, P4, "comment")
84+
ENTRY(cutensorHandleWritePlanCacheToFile, cutensorHandleWritePlanCacheToFile, false, NO_FLAG, P4, "comment")
85+
ENTRY(cutensorReadKernelCacheFromFile, cutensorReadKernelCacheFromFile, false, NO_FLAG, P4, "comment")
86+
ENTRY(cutensorWriteKernelCacheToFile, cutensorWriteKernelCacheToFile, false, NO_FLAG, P4, "comment")
87+
88+
89+
// Logger Functions
90+
ENTRY(cutensorLoggerSetCallback, cutensorLoggerSetCallback, false, NO_FLAG, P4, "comment")
91+
ENTRY(cutensorLoggerSetFile, cutensorLoggerSetFile, false, NO_FLAG, P4, "comment")
92+
ENTRY(cutensorLoggerOpenFile, cutensorLoggerOpenFile, false, NO_FLAG, P4, "comment")
93+
ENTRY(cutensorLoggerSetLevel, cutensorLoggerSetLevel, false, NO_FLAG, P4, "comment")
94+
ENTRY(cutensorLoggerSetMask, cutensorLoggerSetMask, false, NO_FLAG, P4, "comment")
95+
ENTRY(cutensorLoggerForceDisable, cutensorLoggerForceDisable, false, NO_FLAG, P4, "comment")
96+
97+
98+
// cuTENSORMg - General Operations
99+
ENTRY(cutensorMgCreate, cutensorMgCreate, false, NO_FLAG, P4, "comment")
100+
ENTRY(cutensorMgDestroy, cutensorMgDestroy, false, NO_FLAG, P4, "comment")
101+
ENTRY(cutensorMgCreateTensorDescriptor, cutensorMgCreateTensorDescriptor, false, NO_FLAG, P4, "comment")
102+
ENTRY(cutensorMgDestroyTensorDescriptor, cutensorMgDestroyTensorDescriptor, false, NO_FLAG, P4, "comment")
103+
104+
105+
// cuTENSORMg - Copy Operations
106+
ENTRY(cutensorMgCreateCopyDescriptor, cutensorMgCreateCopyDescriptor, false, NO_FLAG, P4, "comment")
107+
ENTRY(cutensorMgDestroyCopyDescriptor, cutensorMgDestroyCopyDescriptor, false, NO_FLAG, P4, "comment")
108+
ENTRY(cutensorMgCopyGetWorkspace, cutensorMgCopyGetWorkspace, false, NO_FLAG, P4, "comment")
109+
ENTRY(cutensorMgCreateCopyPlan, cutensorMgCreateCopyPlan, false, NO_FLAG, P4, "comment")
110+
ENTRY(cutensorMgDestroyCopyPlan, cutensorMgDestroyCopyPlan, false, NO_FLAG, P4, "comment")
111+
ENTRY(cutensorMgCopy, cutensorMgCopy, false, NO_FLAG, P4, "comment")
112+
113+
114+
// cuTENSORMg - Contraction Operations
115+
ENTRY(cutensorMgCreateContractionDescriptor, cutensorMgCreateContractionDescriptor, false, NO_FLAG, P4, "comment")
116+
ENTRY(cutensorMgDestroyContractionDescriptor, cutensorMgDestroyContractionDescriptor, false, NO_FLAG, P4, "comment")
117+
ENTRY(cutensorMgCreateContractionFind, cutensorMgCreateContractionFind, false, NO_FLAG, P4, "comment")
118+
ENTRY(cutensorMgDestroyContractionFind, cutensorMgDestroyContractionFind, false, NO_FLAG, P4, "comment")
119+
ENTRY(cutensorMgContractionGetWorkspace, cutensorMgContractionGetWorkspace, false, NO_FLAG, P4, "comment")
120+
ENTRY(cutensorMgCreateContractionPlan, cutensorMgCreateContractionPlan, false, NO_FLAG, P4, "comment")
121+
ENTRY(cutensorMgDestroyContractionPlan, cutensorMgDestroyContractionPlan, false, NO_FLAG, P4, "comment")
122+
ENTRY(cutensorMgContraction, cutensorMgContraction, false, NO_FLAG, P4, "comment")
123+
124+
// clang-format on

0 commit comments

Comments
 (0)