diff --git a/torch/utils/hipify/cuda_to_hip_mappings.py b/torch/utils/hipify/cuda_to_hip_mappings.py index c6798263f997..6d98fb747280 100644 --- a/torch/utils/hipify/cuda_to_hip_mappings.py +++ b/torch/utils/hipify/cuda_to_hip_mappings.py @@ -607,9 +607,12 @@ CUDA_INCLUDE_MAP = collections.OrderedDict( "channel_descriptor.h", ("hip/channel_descriptor.h", CONV_INCLUDE, API_RUNTIME), ), - ("device_functions.h", ("hip/device_functions.h", CONV_INCLUDE, API_RUNTIME)), - ("driver_types.h", ("hip/driver_types.h", CONV_INCLUDE, API_RUNTIME)), - ("library_types.h", ("hip/library_types.h", CONV_INCLUDE, API_RUNTIME)), + ('include "device_functions.h', ('include "hip/device_functions.h', CONV_INCLUDE, API_RUNTIME)), + ('include