Output of BertEmbeddings

I want to do some changes in the BertModel class based upon my use case. As such I wanted to look at some of the outputs in general for better understanding.
When I tried printing the BertEmbeddings output inside BertModel, it is giving me an illegal memory access RuntimeError. I have attached the screenshot below.

Kindly help me correct this error.
PS: When I tried printing the shape, it was giving me correct output of (batchsize, 512, 768)

The error was basically because of one of my predefined weight tensor was not on cuda. Now everything is working fine. Thanks.