xref: /aosp_15_r20/external/pytorch/torch/utils/hipify/constants.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1"""Constants for annotations in the mapping.
2
3The constants defined here are used to annotate the mapping tuples in cuda_to_hip_mappings.py.
4They are based on
5https://github.com/ROCm/HIPIFY/blob/master/src/Statistics.h
6and fall in three categories: 1) type of mapping, 2) API of mapping, 3) unsupported
7mapping.
8"""
9
10CONV_VERSION = 0,
11CONV_INIT = 1
12CONV_DEVICE = 2
13CONV_MEM = 3
14CONV_KERN = 4
15CONV_COORD_FUNC = 5
16CONV_MATH_FUNC = 6
17CONV_DEVICE_FUNC = 7
18CONV_SPECIAL_FUNC = 8
19CONV_STREAM = 9
20CONV_EVENT = 10
21CONV_OCCUPANCY = 11
22CONV_CONTEXT = 12
23CONV_PEER = 13
24CONV_MODULE = 14
25CONV_CACHE = 15
26CONV_EXEC = 16
27CONV_ERROR = 17
28CONV_DEF = 18
29CONV_TEX = 19
30CONV_GL = 20
31CONV_GRAPHICS = 21
32CONV_SURFACE = 22
33CONV_JIT = 23
34CONV_D3D9 = 24
35CONV_D3D10 = 25
36CONV_D3D11 = 26
37CONV_VDPAU = 27
38CONV_EGL = 28
39CONV_THREAD = 29
40CONV_OTHER = 30
41CONV_INCLUDE = 31
42CONV_INCLUDE_CUDA_MAIN_H = 32
43CONV_TYPE = 33
44CONV_LITERAL = 34
45CONV_NUMERIC_LITERAL = 35
46CONV_LAST = 36
47
48API_DRIVER = 37
49API_RUNTIME = 38
50API_BLAS = 39
51API_SPECIAL = 40
52API_RAND = 41
53API_LAST = 42
54API_FFT = 43
55API_RTC = 44
56API_ROCTX = 45
57
58HIP_UNSUPPORTED = 46
59API_PYTORCH = 1337
60API_CAFFE2 = 1338
61API_C10 = 1339
62API_ROCMSMI = 1340
63