Make adapter info device matching more flexible

Previously all of the device IDs I was aware of for a given vendor could
be comfortably filtered with a single mask, but there's at least one
exception that has come up since that indicates that support for
multiple different masks per vendor is necessary.

This change allows devices to be identified in groups, with each group
being given a different mask or no mask at all. Architectures may be
shared between groups.

Also added the ability to mark a device group as internal, which will
generate the helper functions (like `IsVendorArchitecture()`) for the
internal architectures but not allow them to be exposed in
GPUAdapterInfo. Internal device IDs may overlap with non-internal ones.

Finally, added some validation logic to prevent duplicate deviceIds or
conflicts between device sets/architectures. This was actually supposed
to be in an earlier CL but somehow got omitted from the version that
eventually landed.

Bug: dawn:1498
Change-Id: Icb8bfbee47324cbd9791f63089877ace86c763db
Reviewed-on: https://dawn-review.googlesource.com/c/dawn/+/96900
Reviewed-by: Yunchao He <yunchao.he@intel.com>
Commit-Queue: Brandon Jones <bajones@chromium.org>
Kokoro: Kokoro <noreply+kokoro@google.com>
Reviewed-by: Austin Eng <enga@chromium.org>
Reviewed-by: Kai Ninomiya <kainino@chromium.org>
This commit is contained in:
Brandon Jones 2022-08-02 22:14:35 +00:00 committed by Dawn LUCI CQ
parent ce6246502c
commit 462f648896
4 changed files with 322 additions and 95 deletions

View File

@ -19,6 +19,12 @@ from collections import namedtuple
from generator_lib import Generator, run_generator, FileRender from generator_lib import Generator, run_generator, FileRender
def parse_mask(mask):
if mask:
return int(mask, 0)
return 0xffffffff
class Name: class Name:
def __init__(self, name): def __init__(self, name):
self.name = name self.name = name
@ -59,45 +65,136 @@ class Name:
class Architecture: class Architecture:
def __init__(self, name, json_data): def __init__(self, name, json_data, mask):
self.name = Name(name) self.name = Name(name)
self.devices = [] self.devices = []
mask_num = parse_mask(mask)
for device in json_data: for device in json_data:
device_num = int(device, 0)
# Don't allow duplicate entries
assert device not in self.devices, 'Architecture "{}" contained duplicate deviceID "{}"'.format(
self.name.get(), device)
# Ensure that all device IDs don't contain bits outside the mask
assert device_num & mask_num == device_num, 'Architecture "{}" contained deviceID "{}" which doesn\'t match the given mask of "{}"'.format(
self.name.get(), device, mask)
self.devices.append(device) self.devices.append(device)
class DeviceSet:
def __init__(self, json_data):
self.mask = None
self.internal = False
if 'mask' in json_data:
self.mask = json_data['mask']
if 'internal' in json_data:
self.internal = json_data['internal']
self.architectures = []
if 'architecture' in json_data:
for (arch_name, arch_data) in json_data['architecture'].items():
# Skip any entries that start with an underscore. Used for comments.
if arch_name[0] == '_':
continue
architecture = Architecture(arch_name, arch_data, self.mask)
# Validate that deviceIDs are only allowed to be in one Architecture at a time
for other_architecture in self.architectures:
for device in architecture.devices:
assert device not in other_architecture.devices, 'Architectures "{}" and "{}" both contain deviceID "{}"'.format(
architecture.name.get(),
other_architecture.name.get(), device)
self.architectures.append(architecture)
def validate_devices(self, other_devices, other_mask):
combined_mask = parse_mask(self.mask) & parse_mask(other_mask)
for other_device in other_devices:
other_device_num = int(other_device, 0) & combined_mask
for architecture in self.architectures:
for device in architecture.devices:
device_num = int(device, 0) & combined_mask
assert device_num != other_device_num, 'DeviceID "{}" & mask "{}" conflicts with deviceId "{}" & mask "{}" in architecture "{}"'.format(
other_device, other_mask, device, self.mask,
architecture.name.get())
def maskDeviceId(self):
if not self.mask:
return ''
return ' & ' + self.mask
class Vendor: class Vendor:
def __init__(self, name, json_data): def __init__(self, name, json_data):
self.name = Name(name) self.name = Name(name)
self.id = json_data['id'] self.id = json_data['id']
self.deviceMask = None architecture_dict = {}
if 'deviceMask' in json_data: internal_architecture_dict = {}
self.deviceMask = json_data['deviceMask']
self.architectures = [] self.device_sets = []
if 'devices' in json_data:
for device_data in json_data['devices']:
device_set = DeviceSet(device_data)
if 'architecture' in json_data: for architecture in device_set.architectures:
for (arch_name, arch_data) in json_data['architecture'].items():
# Skip any entries that start with an underscore. Used for comments.
if arch_name[0] == '_':
continue
self.architectures.append(Architecture(arch_name, arch_data)) # Validate that deviceIDs are unique across device sets
for other_device_set in self.device_sets:
# Only validate device IDs between internal and public device sets.
if other_device_set.internal == device_set.internal:
assert device_set.mask != other_device_set.mask, 'Vendor "{}" contained duplicate device masks "{}"'.format(
self.name.get(), device_set.mask)
other_device_set.validate_devices(
architecture.devices, device_set.mask)
def maskDeviceId(self): # Validate that architecture names are unique between internal and public device sets.
if not self.deviceMask: else:
return '' for other_architecture in other_device_set.architectures:
return ' & ' + self.deviceMask assert architecture.name.canonical_case(
) != other_architecture.name.canonical_case(
), '"{}" is defined as both an internal and public architecture'.format(
architecture.name.get())
if device_set.internal:
internal_architecture_dict[
architecture.name.canonical_case(
)] = architecture.name
else:
architecture_dict[architecture.name.canonical_case(
)] = architecture.name
self.device_sets.append(device_set)
# List of unique architecture names under this vendor
self.architecture_names = architecture_dict.values()
self.internal_architecture_names = internal_architecture_dict.values()
def parse_json(json): def parse_json(json):
vendors = [] vendors = []
internal_architecture_count = 0
for (vendor, vendor_data) in json['vendors'].items(): for (vendor_name, vendor_data) in json['vendors'].items():
vendors.append(Vendor(vendor, vendor_data)) # Skip vendors that have a leading underscore. Those are intended to be "comments".
if vendor_name[0] == '_':
continue
return {'vendors': vendors} vendor = Vendor(vendor_name, vendor_data)
vendors.append(vendor)
internal_architecture_count += len(vendor.internal_architecture_names)
return {
'vendors': vendors,
'has_internal': internal_architecture_count > 0
}
class DawnGpuInfoGenerator(Generator): class DawnGpuInfoGenerator(Generator):

View File

@ -28,8 +28,8 @@ namespace {
enum class Architecture { enum class Architecture {
Unknown, Unknown,
{% for vendor in vendors %} {% for vendor in vendors %}
{% for architecture in vendor.architectures %} {% for architecture_name in vendor.architecture_names %}
{{vendor.name.CamelCase()}}_{{architecture.name.CamelCase()}}, {{vendor.name.CamelCase()}}_{{architecture_name.CamelCase()}},
{% endfor %} {% endfor %}
{% endfor %} {% endfor %}
}; };
@ -37,17 +37,20 @@ enum class Architecture {
Architecture GetArchitecture(PCIVendorID vendorId, PCIDeviceID deviceId) { Architecture GetArchitecture(PCIVendorID vendorId, PCIDeviceID deviceId) {
switch(vendorId) { switch(vendorId) {
{% for vendor in vendors %} {% for vendor in vendors %}
{% if len(vendor.architectures) %} {% if len(vendor.device_sets) %}
case kVendorID_{{vendor.name.CamelCase()}}: { case kVendorID_{{vendor.name.CamelCase()}}: {
switch (deviceId{{vendor.maskDeviceId()}}) { {% for device_set in vendor.device_sets %}
{% for architecture in vendor.architectures %} {% if not device_set.internal %}
switch (deviceId{{device_set.maskDeviceId()}}) {
{% for architecture in device_set.architectures %}
{% for device in architecture.devices %} {% for device in architecture.devices %}
case {{device}}: case {{device}}:
{% endfor %} {% endfor %}
return Architecture::{{vendor.name.CamelCase()}}_{{architecture.name.CamelCase()}}; return Architecture::{{vendor.name.CamelCase()}}_{{architecture.name.CamelCase()}};
{% endfor %} {% endfor %}
} }
{% endif %}
{% endfor %}
} break; } break;
{% endif %} {% endif %}
{% endfor %} {% endfor %}
@ -56,6 +59,44 @@ Architecture GetArchitecture(PCIVendorID vendorId, PCIDeviceID deviceId) {
return Architecture::Unknown; return Architecture::Unknown;
} }
{% if has_internal %}
enum class InternalArchitecture {
Unknown,
{% for vendor in vendors %}
{% for architecture_name in vendor.internal_architecture_names %}
{{vendor.name.CamelCase()}}_{{architecture_name.CamelCase()}},
{% endfor %}
{% endfor %}
};
InternalArchitecture GetInternalArchitecture(PCIVendorID vendorId, PCIDeviceID deviceId) {
switch(vendorId) {
{% for vendor in vendors %}
{% if len(vendor.device_sets) %}
case kVendorID_{{vendor.name.CamelCase()}}: {
{% for device_set in vendor.device_sets %}
{% if device_set.internal %}
switch (deviceId{{device_set.maskDeviceId()}}) {
{% for architecture in device_set.architectures %}
{% for device in architecture.devices %}
case {{device}}:
{% endfor %}
return InternalArchitecture::{{vendor.name.CamelCase()}}_{{architecture.name.CamelCase()}};
{% endfor %}
}
{% endif %}
{% endfor %}
} break;
{% endif %}
{% endfor %}
}
return InternalArchitecture::Unknown;
}
{% endif %}
} // namespace } // namespace
// Vendor checks // Vendor checks
@ -68,11 +109,16 @@ Architecture GetArchitecture(PCIVendorID vendorId, PCIDeviceID deviceId) {
// Architecture checks // Architecture checks
{% for vendor in vendors %} {% for vendor in vendors %}
{% if len(vendor.architectures) %} {% if len(vendor.architecture_names) %}
// {{vendor.name.get()}} architectures // {{vendor.name.get()}} architectures
{% for architecture in vendor.architectures %} {% for architecture_name in vendor.architecture_names %}
bool Is{{vendor.name.CamelCase()}}{{architecture.name.CamelCase()}}(PCIVendorID vendorId, PCIDeviceID deviceId) { bool Is{{vendor.name.CamelCase()}}{{architecture_name.CamelCase()}}(PCIVendorID vendorId, PCIDeviceID deviceId) {
return GetArchitecture(vendorId, deviceId) == Architecture::{{vendor.name.CamelCase()}}_{{architecture.name.CamelCase()}}; return GetArchitecture(vendorId, deviceId) == Architecture::{{vendor.name.CamelCase()}}_{{architecture_name.CamelCase()}};
}
{% endfor %}
{% for architecture_name in vendor.internal_architecture_names %}
bool Is{{vendor.name.CamelCase()}}{{architecture_name.CamelCase()}}(PCIVendorID vendorId, PCIDeviceID deviceId) {
return GetInternalArchitecture(vendorId, deviceId) == InternalArchitecture::{{vendor.name.CamelCase()}}_{{architecture_name.CamelCase()}};
} }
{% endfor %} {% endfor %}
{% endif %} {% endif %}
@ -95,9 +141,9 @@ std::string GetArchitectureName(PCIVendorID vendorId, PCIDeviceID deviceId) {
case Architecture::Unknown: case Architecture::Unknown:
return ""; return "";
{% for vendor in vendors %} {% for vendor in vendors %}
{% for architecture in vendor.architectures %} {% for architecture_name in vendor.architecture_names %}
case Architecture::{{vendor.name.CamelCase()}}_{{architecture.name.CamelCase()}}: case Architecture::{{vendor.name.CamelCase()}}_{{architecture_name.CamelCase()}}:
return "{{architecture.name.js_enum_case()}}"; return "{{architecture_name.js_enum_case()}}";
{% endfor %} {% endfor %}
{% endfor %} {% endfor %}
} }

View File

@ -35,11 +35,14 @@ namespace gpu_info {
// Architecture checks // Architecture checks
{% for vendor in vendors %} {% for vendor in vendors %}
{% if len(vendor.architectures) %} {% if len(vendor.architecture_names) %}
// {{vendor.name.get()}} architectures // {{vendor.name.get()}} architectures
{% for architecture in vendor.architectures %} {% for architecture_name in vendor.architecture_names %}
bool Is{{vendor.name.CamelCase()}}{{architecture.name.CamelCase()}}(PCIVendorID vendorId, PCIDeviceID deviceId); bool Is{{vendor.name.CamelCase()}}{{architecture_name.CamelCase()}}(PCIVendorID vendorId, PCIDeviceID deviceId);
{% endfor %}
{% for architecture_name in vendor.internal_architecture_names %}
bool Is{{vendor.name.CamelCase()}}{{architecture_name.CamelCase()}}(PCIVendorID vendorId, PCIDeviceID deviceId);
{% endfor %} {% endfor %}
{% endif %} {% endif %}
{% endfor %} {% endfor %}

View File

@ -17,19 +17,86 @@
"vendors": { "vendors": {
"_Example Vendor": {
"_comment": [
"Each vendor must have an `id` field, which is it's PCI Vendor ID, and may optionally ",
"have a `devices` field which defines device -> architecture mappings which will be ",
"when populating the GPUAdapterInfo. Keys that begin with an `_` (like this one) denote ",
"commnents, and will be ignored by the parser."
],
"id": "0xDEAD",
"devices": [{
"_comment": [
"The `devices` is an array of 'Device Sets', each of which contains an `architecture` ",
"object which defines an architecture name as a key followed by an array of ",
"PCI device IDs that map to that architecture. The architecture name is what will be ",
"reported to the GPUAdapterInfo, made lower-case and with spaces converted to hyphens."
],
"architecture": {
"Alpha": ["0x0A01", "0x0A02"],
"Beta": ["0x0B01", "0x0B02", "0x0B03"]
}
}, {
"_comment": [
"A Device Set can also define a binary `mask`, which will be be applied to any device ",
"ID prior to comparison, making it easier to capture ranges of device IDs. ",
"Architectures may be duplicated between Device Sets as long as the device ID values ",
"Don't overlap."
],
"mask": "0xFF00",
"architecture": {
"Beta": ["0x1B00"],
"Gamma": ["0x0C00", "0x1C00"]
}
}, {
"_comment": [
"Finally, Device Sets may be flagged as `internal`. Architectures defined in an ",
"internal device set will never be reported in the GPUAdapterInfo, but will generate ",
"appropriate helper functions in Dawn (such as `IsExampleVendorDelta()`) in order to ",
"aid in applying workarounds. Device IDs of internal device sets may overlap with ",
"ones defined in non-internal device sets, but architectures must be unique between ",
"internal and non-internal device sets.",
"Internal architectures facilitate the (hopefully rare) cases where more targeted ",
"GPU identification is required by Dawn's implementation than is provided by the normal ",
"architecture groupings. When possible, however, using non-internal architectures ",
"should be preferred when applying workarounds."
],
"internal": true,
"mask": "0xFFF0",
"architecture": {
"Beta Rev 3": ["0x1B30"]
}
}]
},
"AMD": { "AMD": {
"id": "0x1002", "id": "0x1002",
"deviceMask": "0xFFF0", "devices": [{
"mask": "0xFF00",
"architecture": {
"GCN 2": ["0x1300"],
"GCN 5": ["0x1500", "0x1600"],
"RDNA 2": ["0x7400"]
}
}, {
"mask": "0xFFF0",
"architecture": { "architecture": {
"GCN 1": ["0x6600", "0x6610", "0x6660", "0x6790", "0x6800", "0x6810", "0x6820", "0x6830"], "GCN 1": ["0x6600", "0x6610", "0x6660", "0x6790", "0x6800", "0x6810", "0x6820", "0x6830"],
"GCN 2": ["0x1300", "0x1310", "0x6640", "0x6650", "0x67A0", "0x67B0", "0x9830", "0x9850"], "GCN 2": ["0x6640", "0x6650", "0x67A0", "0x67B0", "0x9830", "0x9850"],
"GCN 3": ["0x6900", "0x6920", "0x6930", "0x7300", "0x9870", "0x98E0"], "GCN 3": ["0x6900", "0x6920", "0x6930", "0x7300", "0x9870", "0x98E0"],
"GCN 4": ["0x67C0", "0x67D0", "0x67E0", "0x67F0", "0x6980", "0x6990"], "GCN 4": ["0x67C0", "0x67D0", "0x67E0", "0x67F0", "0x6980", "0x6990"],
"GCN 5": ["0x15D0", "0x1630", "0x1640", "0x66A0", "0x6860", "0x6870", "0x6940", "0x69A0"], "GCN 5": ["0x66A0", "0x6860", "0x6870", "0x6940", "0x69A0"],
"RDNA 1": ["0x7310", "0x7340", "0x7360"], "RDNA 1": ["0x7310", "0x7340", "0x7360"],
"RDNA 2": ["0x73A0", "0x73B0", "0x73D0", "0x73E0", "0x73F0", "0x7420", "0x7430"] "RDNA 2": ["0x73A0", "0x73B0", "0x73D0", "0x73E0", "0x73F0"]
} }
}]
}, },
"Apple": { "Apple": {
@ -47,7 +114,8 @@
"ARM": { "ARM": {
"id": "0x13B5", "id": "0x13B5",
"deviceMask": "0xF0000000", "devices": [{
"mask": "0xF0000000",
"architecture": { "architecture": {
"_comment": [ "_comment": [
"The Midgard GPUs have device IDs like 0x07______ and 0x08______, but it's easiest to", "The Midgard GPUs have device IDs like 0x07______ and 0x08______, but it's easiest to",
@ -59,14 +127,17 @@
"Bifrost": ["0x60000000", "0x70000000"], "Bifrost": ["0x60000000", "0x70000000"],
"Valhall": ["0x90000000", "0xA0000000"] "Valhall": ["0x90000000", "0xA0000000"]
} }
}]
}, },
"Google": { "Google": {
"id": "0x1AE0", "id": "0x1AE0",
"devices": [{
"architecture": { "architecture": {
"Swiftshader": ["0xC0DE"] "Swiftshader": ["0xC0DE"]
} }
}]
}, },
"Img Tec": { "Img Tec": {
@ -76,7 +147,8 @@
"Intel": { "Intel": {
"id": "0x8086", "id": "0x8086",
"deviceMask": "0xFF00", "devices": [{
"mask": "0xFF00",
"architecture": { "architecture": {
"Gen 7": ["0x0100", "0x0400", "0x0A00", "0x0D00", "0x0F00"], "Gen 7": ["0x0100", "0x0400", "0x0A00", "0x0D00", "0x0F00"],
"Gen 8": ["0x1600", "0x2200"], "Gen 8": ["0x1600", "0x2200"],
@ -85,6 +157,7 @@
"Gen 12 LP": ["0x4600", "0x4C00", "0x4900", "0x9A00"], "Gen 12 LP": ["0x4600", "0x4C00", "0x4900", "0x9A00"],
"Gen 12 HP": ["0x4F00", "0x5600"] "Gen 12 HP": ["0x4F00", "0x5600"]
} }
}]
}, },
"Mesa": { "Mesa": {
@ -94,15 +167,18 @@
"Microsoft": { "Microsoft": {
"id": "0x1414", "id": "0x1414",
"devices": [{
"architecture": { "architecture": {
"WARP": ["0x8c"] "WARP": ["0x8c"]
} }
}]
}, },
"Nvidia": { "Nvidia": {
"id": "0x10DE", "id": "0x10DE",
"deviceMask": "0xFF00", "devices": [{
"mask": "0xFF00",
"architecture": { "architecture": {
"Fermi": ["0x0D00"], "Fermi": ["0x0D00"],
"Kepler": ["0x0F00", "0x1000", "0x1100", "0x1200"], "Kepler": ["0x0F00", "0x1000", "0x1100", "0x1200"],
@ -111,26 +187,31 @@
"Turing": ["0x1E00", "0x1F00", "0x2100"], "Turing": ["0x1E00", "0x1F00", "0x2100"],
"Ampere": ["0x2200", "0x2400", "0x2500"] "Ampere": ["0x2200", "0x2400", "0x2500"]
} }
}]
}, },
"Qualcomm": { "Qualcomm": {
"id": "0x5143", "id": "0x5143",
"deviceMask": "0xFF000000", "devices": [{
"mask": "0xFF000000",
"architecture": { "architecture": {
"Adreno 4xx": ["0x04000000"], "Adreno 4xx": ["0x04000000"],
"Adreno 5xx": ["0x05000000"], "Adreno 5xx": ["0x05000000"],
"Adreno 6xx": ["0x06000000"], "Adreno 6xx": ["0x06000000"],
"Adreno 7xx": ["0x07000000"] "Adreno 7xx": ["0x07000000"]
} }
}]
}, },
"Samsung": { "Samsung": {
"id": "0x144d", "id": "0x144d",
"devices": [{
"architecture": { "architecture": {
"RDNA 2": ["0x73A0"] "RDNA 2": ["0x73A0"]
} }
}]
} }
} }