Sync updates from hipify_torch. (#1168)

wenchenvincent · jithunnair-amd · pruthvistony · commit 21a58bce4c7f · 2023-12-05T12:45:26.000-08:00
Co-authored-by: Jithun Nair &lt;37884920+jithunnair-amd@users.noreply.github.com&gt;
diff --git a/torch/utils/hipify/constants.py b/torch/utils/hipify/constants.py
@@ -59,4 +59,5 @@
 API_PYTORCH = 1337
 API_CAFFE2 = 1338
 API_C10 = 1339
-API_ROCMSMI = 1340
+API_PYT_EXT = 1340
+API_ROCMSMI = 1341
diff --git a/torch/utils/hipify/cuda_to_hip_mappings.py b/torch/utils/hipify/cuda_to_hip_mappings.py
@@ -649,6 +649,7 @@
         ("cub/device/device_select.cuh", ("hipcub/hipcub.hpp", CONV_INCLUDE, API_BLAS)),
         ("nvtx3/nvToolsExt.h", ("roctracer/roctx.h", CONV_INCLUDE, API_ROCTX)),
         ("nvml.h", ("rocm_smi/rocm_smi.h", CONV_INCLUDE, API_ROCMSMI)),
+        ("tensorpipe/tensorpipe_cuda.h", ("tensorpipe/tensorpipe_hip.h", CONV_INCLUDE, API_PYT_EXT)),
     ]
 )
 

Original file line number	Diff line number	Diff line change
`@@ -649,6 +649,7 @@`
`649`	`649`	`("cub/device/device_select.cuh", ("hipcub/hipcub.hpp", CONV_INCLUDE, API_BLAS)),`
`650`	`650`	`("nvtx3/nvToolsExt.h", ("roctracer/roctx.h", CONV_INCLUDE, API_ROCTX)),`
`651`	`651`	`("nvml.h", ("rocm_smi/rocm_smi.h", CONV_INCLUDE, API_ROCMSMI)),`
	`652`	`+ ("tensorpipe/tensorpipe_cuda.h", ("tensorpipe/tensorpipe_hip.h", CONV_INCLUDE, API_PYT_EXT)),`
`652`	`653`	`]`
`653`	`654`	`)`
`654`	`655`