We are trying to have caffe2 compiled on AIX. We were able to successfully compile it and run a few caffe2 sample code. But,a RNN sample script fails with an exception as shown below. Any pointers on the cause of this will be helpful.
python char_rnn.py --train_data input.txt
WARNING:root:This caffe2 python run does not have GPU support. Will run in CPU only mode.
WARNING:root:Debug message: No module named caffe2_pybind11_state_hip
Input has 65 characters. Total input size: 1115394
Entering interactive debugger. Type “bt” to print the full stacktrace. Type “help” to see command listing.
Exception when creating gradient for [LSTMUnit]:[enforce fail at pybind_state.cc:523] caffe2::GradientRegistry()->Has(def.type()). .
-> format(op.type, e, str(op))