Skip to content

Commit 2d084a4

Browse files
author
Jin Z
committed
The vendor IDs for AMD and NVIDIA GPUs are 4098 and 4318, respectively
1 parent 81304d4 commit 2d084a4

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

sycl/plugins/hip/pi_hip.cpp

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -980,7 +980,15 @@ pi_result hip_piDeviceGetInfo(pi_device device, pi_device_info param_name,
980980
PI_DEVICE_TYPE_GPU);
981981
}
982982
case PI_DEVICE_INFO_VENDOR_ID: {
983-
return getInfo(param_value_size, param_value, param_value_size_ret, 4098u);
983+
#if defined(__HIP_PLATFORM_AMD__)
984+
pi_uint32 vendor_id = 4098u;
985+
#elif defined(__HIP_PLATFORM_NVIDIA__)
986+
pi_uint32 vendor_id = 4318u;
987+
#else
988+
pi_uint32 vendor_id = 0u;
989+
#endif
990+
return getInfo(param_value_size, param_value, param_value_size_ret,
991+
vendor_id);
984992
}
985993
case PI_DEVICE_INFO_MAX_COMPUTE_UNITS: {
986994
int compute_units = 0;

0 commit comments

Comments
 (0)