Commit b3c3821
authored
[Wave] Fix IREE reference failing with wrong signature (iree-org#858)
Sometime on new machine kernel hashes set for the Wave kernel bleeds
into the kernel has option for the iree reference. This makes our
pipeline try to call th Wave kernel again instead of the IREE reference
vmfb. This shows up as invalid signature, since we are trying to call a
inplace vmfb on a non inplace invoke. We can also see that the function
name instead of the iree kernel_type will be `isolated_benchmark`
entry_function = <VmFunction isolated_benchmark(0rrrIII_r), reflection =
{'iree.abi.declaration': 'sync func
@isolated_benchmark(%input...r<?x?xf16>, %input2: tensor<?x?xf32>,
%input3: index, %input4: index, %input5: index) -> (%output0:
tensor<?x?xf32>)'}>
> vm_context.invoke(entry_function, arg_list, ret_list)
E ValueError: Error invoking function:
c/runtime/src/iree/vm/invocation.c:95: INVALID_ARGUMENT; input list and
function mismatch; expected 6 arguments but passed 2
Signed-off-by: Stanley Winata <[email protected]>1 parent e2e178b commit b3c3821
1 file changed
+1
-0
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
202 | 202 | | |
203 | 203 | | |
204 | 204 | | |
| 205 | + | |
205 | 206 | | |
206 | 207 | | |
207 | 208 | | |
| |||
0 commit comments