Is Flex delegate compatible with CUDA?
16 views
Skip to first unread message
Iosif Hamlatzis
Jan 26, 2022, 2:54:27 AM1/26/22
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