Skip to content

Commit b3c3821

Browse files
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

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

iree/turbine/kernel/wave/iree_utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -202,6 +202,7 @@ def generate_iree_ref(
202202

203203
options.func_name = kernel_type
204204
options.inplace = False
205+
options.kernel_hash = None
205206
options.dynamic_symbols_map = {}
206207
compile_and_invoke(
207208
asm,

0 commit comments

Comments
 (0)