Is Flex delegate compatible with CUDA?

15 views
Skip to first unread message

Iosif Hamlatzis

unread,
Jan 26, 2022, 2:54:27 AMJan 26
to TensorFlow Developers
I have created a model with SELECT OPS but if I have CUDA enabled it crashes both under Python and C++, but works fine if CUDA isn't enabled

When CUDA is enabled (and only then) I get an error message 

"op 'ReadVariableOp' output%d tensor#%d shape mismatch for [144] != []" from file "./tensorflow/lite/delegates/flex/kernel.cc" at line 590 and then it goes to file "./tensorflow/lite/delegates/flex/delegate.cc" in method "CopyFromBufferHandle" in line 160
and crashes "memcpy(output->data.raw, t_data.data(), t_data.size());"

Any idea?
Reply all
Reply to author
Forward
0 new messages