Skip to content

Commit 79aea1d

Browse files
committed
Fix driver extension iteration.
1 parent 69c79da commit 79aea1d

File tree

2 files changed

+8
-5
lines changed

2 files changed

+8
-5
lines changed

lib/level-zero/driver.jl

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -100,15 +100,18 @@ function ipc_properties(drv::ZeDriver)
100100
)
101101
end
102102

103-
# FIXME: throws ZE_RESULT_ERROR_UNSUPPORTED_FEATURE
104103
function extension_properties(drv::ZeDriver)
105104
count_ref = Ref{UInt32}(0)
106105
zeDriverGetExtensionProperties(drv, count_ref, C_NULL)
107106

108107
all_props = Vector{ze_driver_extension_properties_t}(undef, count_ref[])
109108
zeDriverGetExtensionProperties(drv, count_ref, all_props)
110109

111-
return [(name=String([props.name[1:findfirst(isequal(0), props.name)-1]...]),
112-
version=Int(props.version),
113-
) for props in all_props[1:count_ref[]]]
110+
extensions = Dict{String,VersionNumber}()
111+
for prop in all_props[1:count_ref[]]
112+
name = String(UInt8[prop.name[1:findfirst(isequal(0), prop.name)-1]...])
113+
version = unmake_version(prop.version)
114+
extensions[name] = version
115+
end
116+
return extensions
114117
end

test/level-zero.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ api_version(drv)
2020

2121
properties(drv)
2222
ipc_properties(drv)
23-
#extension_properties(drv)
23+
extension_properties(drv)
2424

2525
end
2626

0 commit comments

Comments
 (0)