Skip to content

Commit 0fd199f

Browse files
pruthvistonyjithunnair-amd
authored andcommitted
CONSOLIDATED COMMITS: Hipify updates
==================================== Update to hipify mapping Sync updates from hipify_torch. (#1168) Co-authored-by: Jithun Nair <[email protected]> (cherry picked from commit a4b5019)
1 parent f61af1f commit 0fd199f

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
@@ -4,7 +4,7 @@
44
from .constants import (API_BLAS, API_C10, API_CAFFE2, API_DRIVER, API_FFT,
55
API_PYTORCH, API_RAND, API_ROCTX, API_RTC, API_RUNTIME,
66
API_SPECIAL, API_ROCMSMI, CONV_CACHE, CONV_CONTEXT, CONV_D3D9,
7-
CONV_D3D10, CONV_D3D11, CONV_DEF, CONV_DEVICE,
7+
API_PYT_EXT, CONV_D3D10, CONV_D3D11, CONV_DEF, CONV_DEVICE,
88
CONV_DEVICE_FUNC, CONV_EGL, CONV_ERROR, CONV_EVENT,
99
CONV_EXEC, CONV_GL, CONV_GRAPHICS, CONV_INCLUDE,
1010
CONV_INCLUDE_CUDA_MAIN_H, CONV_INIT, CONV_JIT,
@@ -631,6 +631,7 @@
631631
("nvtx3/nvtx3.hpp", ("roctracer/roctx.h", CONV_INCLUDE, API_ROCTX)),
632632
("nvToolsExt.h", ("roctracer/roctx.h", CONV_INCLUDE, API_ROCTX)),
633633
("nvml.h", ("rocm_smi/rocm_smi.h", CONV_INCLUDE, API_ROCMSMI)),
634+
("tensorpipe/tensorpipe_cuda.h", ("tensorpipe/tensorpipe_hip.h", CONV_INCLUDE, API_PYT_EXT)),
634635
]
635636
)
636637

0 commit comments

Comments
 (0)