mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
eaffd98880
Enables the hipSolver backend for ROCm builds -------------------------------------------------------------------------- - Minimum ROCm version requirement - 5.3 - Introduces new macro USE_LINALG_SOLVER the controls enablement of both cuSOLVER and hipSOLVER - Adds hipSOLVER API to hipification process - combines hipSOLVER and hipSPARSE mappings into single SPECIAL map that takes priority among normal mappings - Torch api to be moved to hipsolver backend (as opposed to magma) include: torch.svd(), torch.geqrf(), torch.orgqr(), torch.ormqr() - Will enable 100+ linalg unit tests for ROCm Pull Request resolved: https://github.com/pytorch/pytorch/pull/97370 Approved by: https://github.com/malfet
62 lines
1.2 KiB
Python
62 lines
1.2 KiB
Python
""" Constants for annotations in the mapping.
|
|
The constants defined here are used to annotate the mapping tuples in cuda_to_hip_mappings.py.
|
|
They are based on
|
|
https://github.com/ROCm-Developer-Tools/HIP/blob/master/hipify-clang/src/Statistics.h
|
|
and fall in three categories: 1) type of mapping, 2) API of mapping, 3) unsupported
|
|
mapping.
|
|
"""
|
|
|
|
CONV_VERSION = 0,
|
|
CONV_INIT = 1
|
|
CONV_DEVICE = 2
|
|
CONV_MEM = 3
|
|
CONV_KERN = 4
|
|
CONV_COORD_FUNC = 5
|
|
CONV_MATH_FUNC = 6
|
|
CONV_DEVICE_FUNC = 7
|
|
CONV_SPECIAL_FUNC = 8
|
|
CONV_STREAM = 9
|
|
CONV_EVENT = 10
|
|
CONV_OCCUPANCY = 11
|
|
CONV_CONTEXT = 12
|
|
CONV_PEER = 13
|
|
CONV_MODULE = 14
|
|
CONV_CACHE = 15
|
|
CONV_EXEC = 16
|
|
CONV_ERROR = 17
|
|
CONV_DEF = 18
|
|
CONV_TEX = 19
|
|
CONV_GL = 20
|
|
CONV_GRAPHICS = 21
|
|
CONV_SURFACE = 22
|
|
CONV_JIT = 23
|
|
CONV_D3D9 = 24
|
|
CONV_D3D10 = 25
|
|
CONV_D3D11 = 26
|
|
CONV_VDPAU = 27
|
|
CONV_EGL = 28
|
|
CONV_THREAD = 29
|
|
CONV_OTHER = 30
|
|
CONV_INCLUDE = 31
|
|
CONV_INCLUDE_CUDA_MAIN_H = 32
|
|
CONV_TYPE = 33
|
|
CONV_LITERAL = 34
|
|
CONV_NUMERIC_LITERAL = 35
|
|
CONV_LAST = 36
|
|
|
|
API_DRIVER = 37
|
|
API_RUNTIME = 38
|
|
API_BLAS = 39
|
|
API_SPECIAL = 40
|
|
API_RAND = 41
|
|
API_LAST = 42
|
|
API_FFT = 43
|
|
API_RTC = 44
|
|
API_ROCTX = 45
|
|
|
|
HIP_UNSUPPORTED = 46
|
|
API_PYTORCH = 1337
|
|
API_CAFFE2 = 1338
|
|
API_C10 = 1339
|
|
API_ROCMSMI = 1340
|