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