Hi Tom,
Thanks for responding! I don’t think the fault lies in incorrect embedding access or something like that, because the same code works fine on CPU. And since that is the only line where some device-specific operation is happening, it seems likely that only is the source of the error.
I did try to run my code with blocking kernel launches as
CUDA_LAUNCH_BLOCKING=1 python3 LSTM.py
But strangely enough, nothing happens after I call the forward method. I tried adding print statements, but nothing is output. It seems the process is stuck and I have no option but to kill it.
Are you certain the numpy conversion isn’t problematic?