Skip to content

Commit ddb6267

Browse files
Update to hipify mapping
Sync updates from hipify_torch. (#1168) Co-authored-by: Jithun Nair <[email protected]>
1 parent 2f123fb commit ddb6267

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

torch/utils/hipify/constants.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,4 +59,5 @@
5959
API_PYTORCH = 1337
6060
API_CAFFE2 = 1338
6161
API_C10 = 1339
62-
API_ROCMSMI = 1340
62+
API_PYT_EXT = 1340
63+
API_ROCMSMI = 1341

torch/utils/hipify/cuda_to_hip_mappings.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
from .constants import (API_BLAS, API_C10, API_CAFFE2, API_DRIVER, API_FFT,
44
API_PYTORCH, API_RAND, API_ROCTX, API_RTC, API_RUNTIME,
55
API_SPECIAL, API_ROCMSMI, CONV_CACHE, CONV_CONTEXT, CONV_D3D9,
6-
CONV_D3D10, CONV_D3D11, CONV_DEF, CONV_DEVICE,
6+
API_PYT_EXT, CONV_D3D10, CONV_D3D11, CONV_DEF, CONV_DEVICE,
77
CONV_DEVICE_FUNC, CONV_EGL, CONV_ERROR, CONV_EVENT,
88
CONV_EXEC, CONV_GL, CONV_GRAPHICS, CONV_INCLUDE,
99
CONV_INCLUDE_CUDA_MAIN_H, CONV_INIT, CONV_JIT,
@@ -624,6 +624,7 @@
624624
("nvtx3/nvtx3.hpp", ("roctracer/roctx.h", CONV_INCLUDE, API_ROCTX)),
625625
("nvToolsExt.h", ("roctracer/roctx.h", CONV_INCLUDE, API_ROCTX)),
626626
("nvml.h", ("rocm_smi/rocm_smi.h", CONV_INCLUDE, API_ROCMSMI)),
627+
("tensorpipe/tensorpipe_cuda.h", ("tensorpipe/tensorpipe_hip.h", CONV_INCLUDE, API_PYT_EXT)),
627628
]
628629
)
629630

0 commit comments

Comments
 (0)