shylie/llvm-project
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Offload] Add specifier for the host type (#141635)

Browse Source 
Summary:
We use this sepcial type to indicate a host value, this will be refined
later but for now it's used as a stand-in device for transfers and
queues. It needs a special kind because it is not a device target as the
other ones so we need to differentiate it between a CPU and GPU type.

Fixes: https://github.com/llvm/llvm-project/issues/141436
This commit is contained in:
Joseph Huber 2025-05-28 08:51:14 -05:00 committed by GitHub
parent 06ee672fc5
commit 0ebe5557d9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
5 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
offload/liboffload/API/Device.td
View File

@ -18,6 +18,7 @@ def : Enum {
Etor<"ALL", "Devices of all types">,
Etor<"GPU", "GPU device type">,
Etor<"CPU", "CPU device type">,
Etor<"Host", "Host device type">,
];
}

2
offload/liboffload/include/generated/OffloadAPI.h
View File

@ -320,6 +320,8 @@ typedef enum ol_device_type_t {
OL_DEVICE_TYPE_GPU = 2,
/// CPU device type
OL_DEVICE_TYPE_CPU = 3,
/// Host device type
OL_DEVICE_TYPE_HOST = 4,
/// @cond
OL_DEVICE_TYPE_FORCE_UINT32 = 0x7fffffff
/// @endcond

3
offload/liboffload/include/generated/OffloadPrint.hpp
View File

@ -224,6 +224,9 @@ inline llvm::raw_ostream &operator<<(llvm::raw_ostream &os,
case OL_DEVICE_TYPE_CPU:
os << "OL_DEVICE_TYPE_CPU";
break;
case OL_DEVICE_TYPE_HOST:
os << "OL_DEVICE_TYPE_HOST";
break;
default:
os << "unknown enumerator";
break;

3
offload/liboffload/src/OffloadImpl.cpp
View File

@ -258,7 +258,8 @@ Error olGetDeviceInfoImplDetail(ol_device_handle_t Device,
case OL_DEVICE_INFO_PLATFORM:
return ReturnValue(Device->Platform);
case OL_DEVICE_INFO_TYPE:
return ReturnValue(OL_DEVICE_TYPE_GPU);
return Device == HostDevice() ? ReturnValue(OL_DEVICE_TYPE_HOST)
: ReturnValue(OL_DEVICE_TYPE_GPU);
case OL_DEVICE_INFO_NAME:
return ReturnValue(GetInfo({"Device Name"}).c_str());
case OL_DEVICE_INFO_VENDOR:

7
offload/unittests/OffloadAPI/device/olGetDeviceInfo.cpp
View File

@ -19,6 +19,13 @@ TEST_P(olGetDeviceInfoTest, SuccessType) {
sizeof(ol_device_type_t), &DeviceType));
}
TEST_P(olGetDeviceInfoTest, HostSuccessType) {
ol_device_type_t DeviceType;
ASSERT_SUCCESS(olGetDeviceInfo(Host, OL_DEVICE_INFO_TYPE,
sizeof(ol_device_type_t), &DeviceType));
ASSERT_EQ(DeviceType, OL_DEVICE_TYPE_HOST);
}
TEST_P(olGetDeviceInfoTest, SuccessPlatform) {
ol_platform_handle_t Platform = nullptr;
ASSERT_SUCCESS(olGetDeviceInfo(Device, OL_DEVICE_INFO_PLATFORM,